Skip to content

Commit

Permalink
Applying comments v2
Browse files Browse the repository at this point in the history
  • Loading branch information
pfinashx committed Mar 31, 2022
1 parent f5733cc commit a396e8b
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions external/anomaly/tests/test_ote_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -123,7 +123,7 @@ def get_anomaly_test_action_classes() -> List[Type[BaseOTETestAction]]:
]


class AnomalyDetectionTrainingTestParameters(DefaultOTETestCreationParametersInterface):
class AnomalyClassificationTrainingTestParameters(DefaultOTETestCreationParametersInterface):
def test_case_class(self) -> Type[OTETestCaseInterface]:
return generate_ote_integration_test_case_class(
get_anomaly_test_action_classes()
Expand Down Expand Up @@ -278,12 +278,12 @@ def __call__(self, data_collector: DataCollector, results_prev_stages: OrderedDi
return results


class TestOTEReallifeAnomalyDetection(OTETrainingTestInterface):
class TestOTEReallifeAnomalyClassification(OTETrainingTestInterface):
"""
The main class of running test in this file.
"""
PERFORMANCE_RESULTS = None # it is required for e2e system
helper = OTETestHelper(AnomalyDetectionTrainingTestParameters())
helper = OTETestHelper(AnomalyClassificationTrainingTestParameters())

@classmethod
def get_list_of_tests(cls, usecase: Optional[str] = None):
Expand Down

0 comments on commit a396e8b

Please sign in to comment.