diff --git a/src/sparsezoo/analyze/__init__.py b/src/sparsezoo/analyze_v1/__init__.py similarity index 100% rename from src/sparsezoo/analyze/__init__.py rename to src/sparsezoo/analyze_v1/__init__.py diff --git a/src/sparsezoo/analyze/analysis.py b/src/sparsezoo/analyze_v1/analysis.py similarity index 99% rename from src/sparsezoo/analyze/analysis.py rename to src/sparsezoo/analyze_v1/analysis.py index 55a43457b..96f4e264a 100644 --- a/src/sparsezoo/analyze/analysis.py +++ b/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 b/src/sparsezoo/analyze_v1/cli.py similarity index 100% rename from src/sparsezoo/analyze/cli.py rename to src/sparsezoo/analyze_v1/cli.py diff --git a/src/sparsezoo/analyze/utils/__init__.py b/src/sparsezoo/analyze_v1/utils/__init__.py similarity index 100% rename from src/sparsezoo/analyze/utils/__init__.py rename to src/sparsezoo/analyze_v1/utils/__init__.py diff --git a/src/sparsezoo/analyze/utils/chart.py b/src/sparsezoo/analyze_v1/utils/chart.py similarity index 99% rename from src/sparsezoo/analyze/utils/chart.py rename to src/sparsezoo/analyze_v1/utils/chart.py index 40edee17e..8c873248c 100644 --- a/src/sparsezoo/analyze/utils/chart.py +++ b/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 b/src/sparsezoo/analyze_v1/utils/helpers.py similarity index 100% rename from src/sparsezoo/analyze/utils/helpers.py rename to src/sparsezoo/analyze_v1/utils/helpers.py diff --git a/src/sparsezoo/analyze/utils/models.py b/src/sparsezoo/analyze_v1/utils/models.py similarity index 100% rename from src/sparsezoo/analyze/utils/models.py rename to src/sparsezoo/analyze_v1/utils/models.py diff --git a/tests/sparsezoo/analyze/helpers.py b/tests/sparsezoo/analyze/helpers.py index 2d197fbea..917776e83 100644 --- 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 index 848cf0331..2806bc3bd 100644 --- 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 index 0743e00ed..bf44b53eb 100644 --- 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 index 274146911..bb22d4c11 100644 --- 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 index e393a815a..884941c34 100644 --- 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 index 8e1453453..8ad816338 100644 --- 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,