Skip to content
This repository has been archived by the owner on Jan 21, 2025. It is now read-only.

Commit

Permalink
Merge pull request #175 from konstellation-io/release/v1.3.12
Browse files Browse the repository at this point in the history
Release/v1.3.12
  • Loading branch information
kafkaphoenix authored Dec 18, 2023
2 parents 8fb083a + 3795df3 commit 70bc6d4
Showing 1 changed file with 5 additions and 6 deletions.
11 changes: 5 additions & 6 deletions py-sdk/sdk/sdk/model_registry/model_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,7 @@ def get_model(self, name: str, version: Optional[str] = None) -> Optional[Model]

object_ = None
try:
exist = self._object_exist(self._get_model_path(name))
exist = self._object_exist(name)
if not exist:
self.logger.error(f"model {name} with version {version} not found in model registry")
raise ModelNotFoundError(name, version)
Expand All @@ -159,7 +159,7 @@ def get_model(self, name: str, version: Optional[str] = None) -> Optional[Model]

self.logger.info(f"model {name} successfully retrieved from model registry")
return Model(
name=self._get_model_name(object_.object_name),
name=self._get_model_name(stats.object_name),
version=stats.metadata.get("x-amz-Model_version"),
format=stats.metadata.get("x-amz-Model_format"),
description=stats.metadata.get("x-amz-Model_description"),
Expand Down Expand Up @@ -239,8 +239,7 @@ def list_model_versions(self, name: str) -> list[ModelInfo]:

def delete_model(self, name: str) -> bool:
try:
path = self._get_model_path(name)
exist = self._object_exist(path)
exist = self._object_exist(name)
if not exist:
self.logger.error(f"model {name} does not found in model registry")
return False
Expand All @@ -253,10 +252,10 @@ def delete_model(self, name: str) -> bool:
self.logger.error(f"{error}")
raise error

def _object_exist(self, key: str) -> bool:
def _object_exist(self, name: str) -> bool:
# minio does not have a method to check if an object exists
try:
self.minio_client.stat_object(bucket_name=self.minio_bucket_name, object_name=key)
self.minio_client.stat_object(bucket_name=self.minio_bucket_name, object_name=self._get_model_path(name))
return True
except Exception as error:
if "code: NoSuchKey" in str(error):
Expand Down

0 comments on commit 70bc6d4

Please sign in to comment.