15 lines
334 B
Python
15 lines
334 B
Python
"""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()
|