Silently ignore missing notebooks (#427)

neptune-ai · Mar 2, 2021 · fa34e77 · fa34e77
1 parent e7d2420
commit fa34e77
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/neptune/internal/backends/hosted_neptune_backend.py b/neptune/internal/backends/hosted_neptune_backend.py
@@ -394,11 +394,14 @@ def create_checkpoint(self, notebook_id, jupyter_path, _file=None):
                     CheckpointDTO = self.leaderboard_swagger_client.get_model('CheckpointDTO')
                     return CheckpointDTO.unmarshal(response.json())
         else:
-            NewCheckpointDTO = self.leaderboard_swagger_client.get_model('NewCheckpointDTO')
-            return self.leaderboard_swagger_client.api.createEmptyCheckpoint(
-                notebookId=notebook_id,
-                checkpoint=NewCheckpointDTO(path=jupyter_path)
-            ).response().result
+            try:
+                NewCheckpointDTO = self.leaderboard_swagger_client.get_model('NewCheckpointDTO')
+                return self.leaderboard_swagger_client.api.createEmptyCheckpoint(
+                    notebookId=notebook_id,
+                    checkpoint=NewCheckpointDTO(path=jupyter_path)
+                ).response().result
+            except HTTPNotFound:
+                return None
 
     @with_api_exceptions_handler
     def get_experiment(self, experiment_id):