seekrai 0.2.0__tar.gz → 0.2.2__tar.gz

{seekrai-0.2.0 → seekrai-0.2.2}/PKG-INFO

@@ -1,8 +1,7 @@
 Metadata-Version: 2.1
 Name: seekrai
-Version: 0.2.0
+Version: 0.2.2
 Summary: Python client for SeekrAI
-Home-page: https://gitlab.cb.ntent.com/ml/seekr-py
 License: Apache-2.0
 Author: SeekrFlow
 Author-email: support@seekr.com
@@ -28,7 +27,6 @@ Requires-Dist: tabulate (>=0.9.0,<0.10.0)
 Requires-Dist: tqdm (>=4.66.2,<5.0.0)
 Requires-Dist: typer (>=0.9,<0.13)
 Project-URL: Homepage, https://www.seekr.com/
-Project-URL: Repository, https://gitlab.cb.ntent.com/ml/seekr-py
 Description-Content-Type: text/markdown
 
 The Seekr Python Library is the official Python client for SeekrFlow's API platform, providing a convenient way for interacting with the REST APIs and enables easy integrations with Python 3.8+ applications with easy to use synchronous and asynchronous clients.

{seekrai-0.2.0 → seekrai-0.2.2}/pyproject.toml

@@ -14,7 +14,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "seekrai"
-version = "0.2.0"
+version = "0.2.2"
 authors = [
     "SeekrFlow <support@seekr.com>"
 ]
@@ -26,8 +26,8 @@ classifiers = [
     "License :: OSI Approved :: Apache Software License",
     "Operating System :: POSIX :: Linux",
 ]
-repository = "https://gitlab.cb.ntent.com/ml/seekr-py"
-homepage = "https://gitlab.cb.ntent.com/ml/seekr-py"
+#repository = "https://gitlab.cb.ntent.com/ml/seekr-py"
+#homepage = "https://gitlab.cb.ntent.com/ml/seekr-py"
 
 
 [tool.poetry.dependencies]

{seekrai-0.2.0 → seekrai-0.2.2}/src/seekrai/types/finetune.py

@@ -118,7 +118,6 @@ class InfrastructureConfig(BaseModel):
     accel_type: AcceleratorType
     n_accel: int
 
-
 class FinetuneRequest(BaseModel):
     """
     Fine-tune request type
@@ -141,14 +140,16 @@ class FinetuneResponse(BaseModel):
     # validation_files: str | None = None TODO
     # base model name
     model: str | None = None
+    accel_type: AcceleratorType
+    n_accel: int
     # number of epochs
-    # n_epochs: int | None = None
+    n_epochs: int | None = None
     # number of checkpoints to save
     # n_checkpoints: int | None = None # TODO
     # training batch size
-    # batch_size: int | None = None
+    batch_size: int | None = None
     # training learning rate
-    # learning_rate: float | None = None
+    learning_rate: float | None = None
     # number of steps between evals
     # eval_steps: int | None = None TODO
     # is LoRA finetune boolean
@@ -159,6 +160,8 @@ class FinetuneResponse(BaseModel):
     # created/updated datetime stamps
     created_at: datetime | None = None
     # updated_at: str | None = None
+    # up to 40 character suffix for output model name
+    experiment_name: str | None = None
     # job status
     status: FinetuneJobStatus | None = None