added model io functions
This commit is contained in:
@@ -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()
|
||||
Reference in New Issue
Block a user