Fix unit tests

NVIDIA · Nov 25, 2024 · 94ace50 · 94ace50
1 parent 32a7bd9
commit 94ace50
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/unit_test/apis/analytix_test.py b/tests/unit_test/apis/analytix_test.py
@@ -14,10 +14,9 @@
 
 import pytest
 
-from nvflare.apis.analytix import _DATA_TYPE_KEY, AnalyticsData, AnalyticsDataType
+from nvflare.apis.analytix import _DATA_TYPE_KEY, AnalyticsData, AnalyticsDataType, LogWriterName, TrackConst
 from nvflare.apis.dxo import DXO, DataKind
 from nvflare.apis.utils.analytix_utils import create_analytic_dxo
-from nvflare.app_common.tracking.tracker_types import LogWriterName, TrackConst
 
 FROM_DXO_TEST_CASES = [
     ("hello", 3.0, 1, AnalyticsDataType.SCALAR),

diff --git a/tests/unit_test/app_common/widgets/streaming_test.py b/tests/unit_test/app_common/widgets/streaming_test.py
@@ -16,12 +16,11 @@
 
 import pytest
 
-from nvflare.apis.analytix import AnalyticsDataType
+from nvflare.apis.analytix import AnalyticsDataType, LogWriterName, TrackConst
 from nvflare.apis.dxo import DXO, DataKind
 from nvflare.apis.fl_component import FLComponent
 from nvflare.apis.fl_context import FLContext
 from nvflare.apis.utils.analytix_utils import create_analytic_dxo, send_analytic_dxo
-from nvflare.app_common.tracking.tracker_types import LogWriterName, TrackConst
 
 INVALID_TEST_CASES = [
     (list(), dict(), FLContext(), TypeError, f"expect comp to be an instance of FLComponent, but got {type(list())}"),