Merge pull request #794 from neptune-ai/dev/model-registry

Model registry

.github/actions/lint/action.yml

@@ -10,13 +10,13 @@ runs:
           python-version: 3.9
 
       - name: Install dependencies
-        run: pip install -r requirements.txt -r requirements/test_requirements.txt
+        run: pip install -r requirements.txt -r requirements/e2e_requirements.txt
         shell: bash
 
       - name: Run pre-commit
         run: pre-commit run --all-files
         shell: bash
 
       - name: Lint
-        run: python -m pylint -j 0 -f parseable neptune tests
+        run: python -m pylint -j 0 -f parseable neptune tests e2e_tests
         shell: bash

CHANGELOG.md

@@ -1,3 +1,14 @@
+## [UNRELEASED] neptune-client 0.15.0
+
+## Features
+- Methods for creating and manipulating Model Registry objects ([#794](https://github.com/neptune-ai/neptune-client/pull/794))
+
+### Changes
+- Renamed --run parameter to --object in `neptune sync` (previous kept as deprecated, [#849](https://github.com/neptune-ai/neptune-client/pull/849))
+- More helpful error message on SSL validation problem ([#853](https://github.com/neptune-ai/neptune-client/pull/853))
+- Added names to daemon worker threads ([#851](https://github.com/neptune-ai/neptune-client/pull/851))
+- Stopped forwarding every attribute from Handler to Attribute ([#815](https://github.com/neptune-ai/neptune-client/pull/815))
+
 ## neptune-client 0.14.3
 
 ## Features

alpha_integration_dev/new_client.py

@@ -41,7 +41,9 @@
 class NewClientFeatures(ClientFeatures):
     def __init__(self):
         super().__init__()
-        self.exp = neptune.init(source_files="alpha_integration_dev/*.py")
+        self.exp = neptune.init(
+            source_files="alpha_integration_dev/*.py",
+        )
 
         # download sources
         self.exp.sync()

e2e_tests/base.py

@@ -13,17 +13,16 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-__all__ = [
-    "BaseE2ETest",
-]
+__all__ = ["BaseE2ETest", "AVAILABLE_CONTAINERS", "fake"]
 
-import uuid
 import inspect
 
 from faker import Faker
 
 fake = Faker()
 
+AVAILABLE_CONTAINERS = ["project", "run", "model", "model_version"]
+
 
 class BaseE2ETest:
     def gen_key(self):

e2e_tests/conftest.py

@@ -13,17 +13,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+# pylint: disable=redefined-outer-name
 import os
+import time
 
 from faker import Faker
 import boto3
 import pytest
 
 from neptune.management.internal.utils import normalize_project_name
 from neptune.management import create_project, add_project_member
-import neptune.new as neptune
 
-from e2e_tests.utils import a_project_name, Environment
+from e2e_tests.utils import initialize_container, a_project_name, Environment
 
 fake = Faker()
 
@@ -44,6 +45,8 @@ def environment():
         api_token=admin_token,
     )
 
+    time.sleep(10)
+
     add_project_member(
         name=created_project_identifier,
         username=user,
@@ -64,15 +67,27 @@ def environment():
 
 @pytest.fixture(scope="session")
 def container(request, environment):
-    if request.param == "project":
-        project = neptune.init_project(name=environment.project)
-        yield project
-        project.stop()
-
-    if request.param == "run":
-        exp = neptune.init_run(project=environment.project)
-        yield exp
-        exp.stop()
+    exp = initialize_container(
+        container_type=request.param, project=environment.project
+    )
+    yield exp
+    exp.stop()
+
+
+@pytest.fixture(scope="session")
+def containers_pair(request, environment):
+    container_a_type, container_b_type = request.param.split("-")
+    container_a = initialize_container(
+        container_type=container_a_type, project=environment.project
+    )
+    container_b = initialize_container(
+        container_type=container_b_type, project=environment.project
+    )
+
+    yield container_a, container_b
+
+    container_b.stop()
+    container_a.stop()
 
 
 @pytest.fixture(scope="session")