skip 2xx check on Oras-py Registry.push response

Signed-off-by: Isabella do Amaral <[email protected]>

omlmd/helpers.py

@@ -97,7 +97,9 @@ def push(
                 manifest_config=manifest_cfg,
                 do_chunked=True,
             )
-            self.notify_listeners(PushEvent(result, target, model_metadata))
+            self.notify_listeners(
+                PushEvent.from_response(result, target, model_metadata)
+            )
             return result
         finally:
             if owns_meta_files:

omlmd/listener.py

@@ -28,13 +28,12 @@ class Event(ABC):
 
 @dataclass
 class PushEvent(Event):
-    response: requests.Response
+    digest: str
     target: str
     metadata: ModelMetadata
 
-    @property
-    def ok(self) -> bool:
-        return self.response.status_code == 200
-
-    def get_digest(self) -> str:
-        return self.response.headers["Docker-Content-Digest"] if self.ok else ""
+    @classmethod
+    def from_response(
+        cls, response: requests.Response, target: str, metadata: ModelMetadata
+    ) -> t.Self:
+        return cls(response.headers["Docker-Content-Digest"], target, metadata)