From 31f0387b88626ecf20e7a4eb91dfe6df77accea6 Mon Sep 17 00:00:00 2001
From: Cliff Hodel <hodelcl@ethz.ch>
Date: Wed, 20 Dec 2023 23:51:25 +0100
Subject: [PATCH] fix import statement

---
 dace/sdfg/work_depth_analysis/operational_intensity.py | 3 +--
 tests/sdfg/{work_depth_tests.py => work_depth_test.py} | 0
 2 files changed, 1 insertion(+), 2 deletions(-)
 rename tests/sdfg/{work_depth_tests.py => work_depth_test.py} (100%)

diff --git a/dace/sdfg/work_depth_analysis/operational_intensity.py b/dace/sdfg/work_depth_analysis/operational_intensity.py
index f9c3836e40..f165b3b510 100644
--- a/dace/sdfg/work_depth_analysis/operational_intensity.py
+++ b/dace/sdfg/work_depth_analysis/operational_intensity.py
@@ -17,9 +17,8 @@
 from dace.transformation.pass_pipeline import FixedPointPipeline
 
 from dace.data import Array
-from dace.sdfg.work_depth_analysis.op_in_helpers import CacheLineTracker, AccessStack
+from dace.sdfg.work_depth_analysis.op_in_helpers import CacheLineTracker, AccessStack, fit_curve, plot, compute_mape
 from dace.sdfg.work_depth_analysis.work_depth import analyze_sdfg, get_tasklet_work
-from dace.sdfg.work_depth_analysis.extrapolation import fit_curve, plot, compute_mape
 
 
 class SymbolRange():
diff --git a/tests/sdfg/work_depth_tests.py b/tests/sdfg/work_depth_test.py
similarity index 100%
rename from tests/sdfg/work_depth_tests.py
rename to tests/sdfg/work_depth_test.py