fixed mypy types
CI Pipeline / Test (pull_request) Successful in 2m23s
CI Pipeline / Build and Publish (./Dockerfile.model, ${{ vars.docker_repo_url }}/${{ gitea.repository }}/model) (pull_request) Successful in 7m3s
CI Pipeline / Build and Publish (./Dockerfile.web_ui, ${{ vars.docker_repo_url }}/${{ gitea.repository }}/web_ui) (pull_request) Successful in 6m13s
CI Pipeline / Test (pull_request) Successful in 2m23s
CI Pipeline / Build and Publish (./Dockerfile.model, ${{ vars.docker_repo_url }}/${{ gitea.repository }}/model) (pull_request) Successful in 7m3s
CI Pipeline / Build and Publish (./Dockerfile.web_ui, ${{ vars.docker_repo_url }}/${{ gitea.repository }}/web_ui) (pull_request) Successful in 6m13s
This commit is contained in:
@@ -1,8 +1,9 @@
|
||||
from importlib import import_module
|
||||
from types import ModuleType
|
||||
|
||||
from torch.nn import Module
|
||||
|
||||
|
||||
def get_class(path: str) -> ModuleType:
|
||||
def get_class(path: str) -> Module:
|
||||
parts = path.split('.')
|
||||
module_path = '.'.join(parts[:-1])
|
||||
class_name = parts[-1]
|
||||
|
||||
@@ -5,8 +5,8 @@ from pathlib import Path
|
||||
|
||||
import torch
|
||||
from minio import Minio
|
||||
from src.models import VisualCommunicationModel
|
||||
|
||||
from model.src.models import VisualCommunicationModel
|
||||
from shared.data_store import get_model
|
||||
|
||||
from .get_model_name import get_model_name
|
||||
|
||||
Reference in New Issue
Block a user