diff --git a/hta/common/types.py b/hta/common/types.py index ed594ac6..43db40d8 100644 --- a/hta/common/types.py +++ b/hta/common/types.py @@ -1,16 +1,9 @@ -from dataclasses import dataclass from enum import Enum from typing import Dict import pandas as pd -@dataclass -class OperationOutcome: - success: bool - reason: str - - class DeviceType(Enum): UNKNOWN = 0 CPU = 1 diff --git a/hta/utils/checker.py b/hta/utils/checker.py index 86eef05c..139c8f32 100644 --- a/hta/utils/checker.py +++ b/hta/utils/checker.py @@ -1,7 +1,12 @@ import os import os.path +from dataclasses import dataclass -from hta.common.types import OperationOutcome + +@dataclass +class OperationOutcome: + success: bool + reason: str def is_valid_directory(