docs: Fix SDK and Schema reference guide

Signed-off-by: Dimitris Poulopoulos <[email protected]>
mozilla-ai · Nov 20, 2024 · bbd7806 · bbd7806
1 parent ccb384b
commit bbd7806
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/docs/source/reference/schemas.rst b/docs/source/reference/schemas.rst
@@ -1,14 +1,14 @@
 Schemas
 =======
 
-.. automodule:: schemas.completions
+.. automodule:: lumigator_schemas.completions
    :members:
 
-.. automodule:: schemas.datasets
+.. automodule:: lumigator_schemas.datasets
    :members:
 
-.. automodule:: schemas.jobs
+.. automodule:: lumigator_schemas.jobs
    :members:
 
-.. automodule:: schemas.extras
+.. automodule:: lumigator_schemas.extras
    :members:
diff --git a/docs/source/reference/sdk.rst b/docs/source/reference/sdk.rst
@@ -11,7 +11,7 @@ Lumigator Client
 The main entry point to the SDK is the `LumigatorClient` class. You can create an instance of this
 class by providing the Lumigator API host and your Ray cluster address.
 
-.. automodule:: sdk.lumigator
+.. automodule:: lumigator_sdk.lumigator
    :members:
    :undoc-members:
 
@@ -21,7 +21,7 @@ Health
 The `Health` class provides a simple interface to check the health of the Lumigator API and the
 status of the Ray jobs running on the cluster.
 
-.. automodule:: sdk.health
+.. automodule:: lumigator_sdk.health
    :members:
    :undoc-members:
 
@@ -30,7 +30,7 @@ Datasets
 
 The `Datasets` class provides a simple interface to create, update, delete, and list datasets.
 
-.. automodule:: sdk.lm_datasets
+.. automodule:: lumigator_sdk.lm_datasets
    :members:
    :undoc-members:
 
@@ -40,7 +40,7 @@ Jobs
 The `Jobs` class provides a simple interface to submit and monitor jobs. Currently, we support two
 types of jobs: Inference and Evaluation.
 
-.. automodule:: sdk.jobs
+.. automodule:: lumigator_sdk.jobs
    :members:
    :undoc-members:
 
@@ -50,7 +50,7 @@ Completions
  The `Completions` class provides a simple interface to request completions from external APIs.
  Currently, we support two APIs: OpenAI's and Mistral's.
 
-.. automodule:: sdk.completions
+.. automodule:: lumigator_sdk.completions
    :members:
    :undoc-members:
 
@@ -60,6 +60,6 @@ Base Client
 The `BaseClient` class provides a base class for the LumigatorClient. You can use this class to
 create your own client with custom methods.
 
-.. automodule:: sdk.client
+.. automodule:: lumigator_sdk.client
    :members:
    :undoc-members: