added model io functions

This commit is contained in:
Brian Bjarke Jensen
2024-07-14 11:15:56 +02:00
parent ff88dc1719
commit a12f8c46e8
3 changed files with 60 additions and 0 deletions
+14
View File
@@ -0,0 +1,14 @@
"""Definition of get_model function."""
from minio import Minio
from torch.nn import Module
def get_model(
client: Minio,
object_name: str,
) -> Module:
"""Get model from model subfolder in bucket in Minio."""
assert isinstance(client, Minio)
assert isinstance(object_name, str)
raise NotImplementedError()