From 4b164bc470278edd160b7b9b5059a32c4822b376 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adam=20Nie=C5=BCurawski?= <aniezurawski@gmail.com>
Date: Thu, 12 May 2022 09:21:11 +0200
Subject: [PATCH] Validate model key and fix error handling (#896)

* Validate model key and fix error handling

* Fix

* Add commment

* Fix

* Fix

* Changelog
---
 CHANGELOG.md                                            | 8 +++++++-
 neptune/new/internal/backends/hosted_neptune_backend.py | 2 ++
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 6af5cd7c3..7d7acb3c6 100644
--- 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
index 8adfe6fdb..a763ddb7e 100644
--- 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: