From b44ac5aed31bebda1e8cdab6ebd06daad51f153d Mon Sep 17 00:00:00 2001 From: John Lambert Date: Thu, 12 Oct 2023 10:37:39 -0400 Subject: [PATCH] Fix imports --- machine/jobs/build_nmt_engine.py | 3 +-- machine/jobs/nmt_engine_build_job.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/machine/jobs/build_nmt_engine.py b/machine/jobs/build_nmt_engine.py index 794f057..cb95765 100644 --- a/machine/jobs/build_nmt_engine.py +++ b/machine/jobs/build_nmt_engine.py @@ -2,11 +2,10 @@ import json import logging import os -from typing import Callable, Optional, cast +from typing import cast from clearml import Task -from ..utils.canceled_error import CanceledError from .clearml_shared_file_service import ClearMLSharedFileService from .config import SETTINGS from .nmt_engine_build_job import NmtEngineBuildJob diff --git a/machine/jobs/nmt_engine_build_job.py b/machine/jobs/nmt_engine_build_job.py index 1a7688e..8d30e95 100644 --- a/machine/jobs/nmt_engine_build_job.py +++ b/machine/jobs/nmt_engine_build_job.py @@ -2,8 +2,7 @@ from contextlib import ExitStack from typing import Any, Optional, Sequence -import pandas as pd -from clearml import Model, Task +from clearml import Task from ..corpora.corpora_utils import batch from ..translation.translation_engine import TranslationEngine