Validate model key and fix error handling (#896)

* Validate model key and fix error handling * Fix * Add commment * Fix * Fix * Changelog
neptune-ai · May 12, 2022 · 4b164bc · 4b164bc
1 parent 806d76d
commit 4b164bc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,8 +1,14 @@
-## [UNRELEASED] neptune-client 0.16.2
+## [UNRELEASED] neptune-client 0.16.3
+
+
+## neptune-client 0.16.2
 
 ## Features
  - Sync only offline runs inside '.neptune' directory CLI flag ([#894](https://github.com/neptune-ai/neptune-client/pull/894))
 
+## Fixes
+- Fix handling of server errors ([#896](https://github.com/neptune-ai/neptune-client/pull/896))
+
 ## neptune-client 0.16.1
 
 ## Features

diff --git a/neptune/new/internal/backends/hosted_neptune_backend.py b/neptune/new/internal/backends/hosted_neptune_backend.py
@@ -362,6 +362,7 @@ def create_run(
             additional_params=additional_params,
         )
 
+    @with_api_exceptions_handler
     def create_model(self, project_id: UniqueId, key: str = "") -> ApiExperiment:
         additional_params = {
             "key": key,
@@ -374,6 +375,7 @@ def create_model(self, project_id: UniqueId, key: str = "") -> ApiExperiment:
             additional_params=additional_params,
         )
 
+    @with_api_exceptions_handler
     def create_model_version(
         self, project_id: UniqueId, model_id: UniqueId
     ) -> ApiExperiment: