Added support for family parameter

.pre-commit-config.yaml

@@ -27,7 +27,7 @@ repos:
         args: [ --config-file, pyproject.toml ]
         pass_filenames: false
         additional_dependencies:
-          - neptune-api==0.3.0
+          - neptune-api==0.4.0
           - more-itertools
 default_language_version:
     python: python3

pyproject.toml

@@ -11,7 +11,7 @@ pattern = "default-unprefixed"
 [tool.poetry.dependencies]
 python = "^3.8"
 
-neptune-api = "0.3.0"
+neptune-api = "0.4.0"
 more-itertools = "^10.0.0"
 
 [tool.poetry]

src/neptune_scale/__init__.py

@@ -197,7 +197,7 @@ def _create_run(
                 creation_time=None if creation_time is None else datetime_to_proto(creation_time),
             ),
         )
-        self._backend.submit(operation=operation)
+        self._backend.submit(operation=operation, family=self._family)
         # TODO: Enqueue on the operations queue
         # self._operations_queue.enqueue(operation=operation)
 
@@ -265,6 +265,6 @@ def log(
         )
 
         for operation in splitter:
-            self._backend.submit(operation=operation)
+            self._backend.submit(operation=operation, family=self._family)
             # TODO: Enqueue on the operations queue
             # self._operations_queue.enqueue(operation=operation)

src/neptune_scale/api/api_client.py

@@ -43,8 +43,8 @@ def __init__(self, api_token: str) -> None:
         config, token_urls = get_config_and_token_urls(credentials=credentials)
         self._backend = create_auth_api_client(credentials=credentials, config=config, token_refreshing_urls=token_urls)
 
-    def submit(self, operation: RunOperation) -> None:
-        _ = submit_operation.sync(client=self._backend, body=operation)
+    def submit(self, operation: RunOperation, family: str) -> None:
+        _ = submit_operation.sync(client=self._backend, family=family, body=operation)
 
     def cleanup(self) -> None:
         pass