Rename legacy analyze to analyze_v1 (#460)

neuralmagic · Feb 21, 2024 · 44b7972 · 44b7972
1 parent 64681fe
commit 44b7972
Show file tree

Hide file tree

Showing 13 changed files with 11 additions and 11 deletions.
diff --git a/src/sparsezoo/analyze/__init__.py → src/sparsezoo/analyze_v1/__init__.py b/src/sparsezoo/analyze/__init__.py → src/sparsezoo/analyze_v1/__init__.py
diff --git a/src/sparsezoo/analyze/analysis.py → src/sparsezoo/analyze_v1/analysis.py b/src/sparsezoo/analyze/analysis.py → src/sparsezoo/analyze_v1/analysis.py
@@ -30,8 +30,8 @@
 from pydantic import BaseModel, Field, PositiveFloat, PositiveInt
 
 from sparsezoo import Model
-from sparsezoo.analyze.utils.helpers import numpy_array_representer
-from sparsezoo.analyze.utils.models import (
+from sparsezoo.analyze_v1.utils.helpers import numpy_array_representer
+from sparsezoo.analyze_v1.utils.models import (
     DenseSparseOps,
     Entry,
     ModelEntry,

diff --git a/src/sparsezoo/analyze/cli.py → src/sparsezoo/analyze_v1/cli.py b/src/sparsezoo/analyze/cli.py → src/sparsezoo/analyze_v1/cli.py
diff --git a/src/sparsezoo/analyze/utils/__init__.py → src/sparsezoo/analyze_v1/utils/__init__.py b/src/sparsezoo/analyze/utils/__init__.py → src/sparsezoo/analyze_v1/utils/__init__.py
diff --git a/src/sparsezoo/analyze/utils/chart.py → src/sparsezoo/analyze_v1/utils/chart.py b/src/sparsezoo/analyze/utils/chart.py → src/sparsezoo/analyze_v1/utils/chart.py
@@ -25,7 +25,7 @@
     plt = None
     matplotlib_available = False
 
-from sparsezoo.analyze.analysis import ModelAnalysis, NodeAnalysis
+from sparsezoo.analyze_v1.analysis import ModelAnalysis, NodeAnalysis
 
 
 __all__ = [

diff --git a/src/sparsezoo/analyze/utils/helpers.py → src/sparsezoo/analyze_v1/utils/helpers.py b/src/sparsezoo/analyze/utils/helpers.py → src/sparsezoo/analyze_v1/utils/helpers.py
diff --git a/src/sparsezoo/analyze/utils/models.py → src/sparsezoo/analyze_v1/utils/models.py b/src/sparsezoo/analyze/utils/models.py → src/sparsezoo/analyze_v1/utils/models.py
diff --git a/tests/sparsezoo/analyze/helpers.py b/tests/sparsezoo/analyze/helpers.py
@@ -22,7 +22,7 @@
 import pytest
 
 from sparsezoo import Model
-from sparsezoo.analyze import ModelAnalysis
+from sparsezoo.analyze_v1 import ModelAnalysis
 from sparsezoo.utils.graph_editor import ONNXGraph
 
 

diff --git a/tests/sparsezoo/analyze/test_analysis.py b/tests/sparsezoo/analyze/test_analysis.py
@@ -16,7 +16,7 @@
 
 import pytest
 
-from sparsezoo.analyze import ModelAnalysis
+from sparsezoo.analyze_v1 import ModelAnalysis
 from tests.sparsezoo.analyze.helpers import (
     get_expected_analysis,
     get_generated_analysis,

diff --git a/tests/sparsezoo/analyze/test_analysis/test_analysis_schemas.py b/tests/sparsezoo/analyze/test_analysis/test_analysis_schemas.py
@@ -15,7 +15,7 @@
 import pytest
 import yaml
 
-from sparsezoo.analyze import (
+from sparsezoo.analyze_v1 import (
     BenchmarkResult,
     BenchmarkScenario,
     ImposedSparsificationInfo,

diff --git a/tests/sparsezoo/analyze/test_analysis/test_analysis_summary.py b/tests/sparsezoo/analyze/test_analysis/test_analysis_summary.py
@@ -15,9 +15,9 @@
 
 import pytest
 
-from sparsezoo.analyze import ModelAnalysis
-from sparsezoo.analyze.analysis import ModelAnalysisSummary
-from sparsezoo.analyze.utils.models import ModelEntry, Section, SizedModelEntry
+from sparsezoo.analyze_v1 import ModelAnalysis
+from sparsezoo.analyze_v1.analysis import ModelAnalysisSummary
+from sparsezoo.analyze_v1.utils.models import ModelEntry, Section, SizedModelEntry
 
 
 @pytest.fixture

diff --git a/tests/sparsezoo/analyze/test_model_analysis_creation.py b/tests/sparsezoo/analyze/test_model_analysis_creation.py
@@ -18,7 +18,7 @@
 import pytest
 
 from sparsezoo import Model
-from sparsezoo.analyze import ModelAnalysis
+from sparsezoo.analyze_v1 import ModelAnalysis
 
 
 def onnx_stub():

diff --git a/tests/sparsezoo/analyze/utils/test_chart.py b/tests/sparsezoo/analyze/utils/test_chart.py
@@ -16,7 +16,7 @@
 
 import pytest
 
-from sparsezoo.analyze import (
+from sparsezoo.analyze_v1 import (
     ModelAnalysis,
     draw_operation_chart,
     draw_parameter_chart,