Add line_profiler (#81)

Author: alex98247

pyproject.toml

@@ -24,6 +24,7 @@ jsonschema = "==4.23.0"
 rich = "==13.9.4"
 click = ">=8.2.1"
 dacite = "==1.9.2"
+line_profiler = "5.0.0"
 pysatl-criterion = {path = "./pysatl_criterion"}
 
 [tool.poetry.group.dev.dependencies]

pysatl_experiment/experiment_new/experiment/experiment.py

@@ -1,3 +1,5 @@
+from line_profiler import profile
+
 from pysatl_experiment.experiment_new.experiment_steps.experiment_steps import ExperimentSteps
 from pysatl_experiment.experiment_new.model.experiment_step.experiment_step import IExperimentStep
 
@@ -10,6 +12,7 @@ class Experiment:
     def __init__(self, experiment_steps: ExperimentSteps):
         self.experiment_steps = experiment_steps
 
+    @profile
     def run_experiment(self) -> None:
         """
         Run experiment.

pysatl_experiment/experiment_new/step/execution/common/utils/utils.py

@@ -1,6 +1,9 @@
+from line_profiler import profile
+
 from pysatl_experiment.persistence.model.random_values.random_values import IRandomValuesStorage, RandomValuesCountQuery
 
 
+@profile
 def get_sample_data_from_storage(
     generator_name: str,
     generator_parameters: list[float],

pysatl_experiment/experiment_new/step/execution/critical_value/critical_value.py

@@ -1,5 +1,7 @@
 from dataclasses import dataclass
 
+from line_profiler import profile
+
 from pysatl_criterion.persistence.model.limit_distribution.limit_distribution import (
     ILimitDistributionStorage,
     LimitDistributionModel,
@@ -43,6 +45,7 @@ def __init__(
         self.data_storage = data_storage
         self.result_storage = result_storage
 
+    @profile
     def run(self) -> None:
         """
         Run standard critical value execution step.
@@ -72,6 +75,7 @@ def run(self) -> None:
                 results_statistics=results_statistics,
             )
 
+    @profile
     def _save_result_to_storage(
         self,
         experiment_id: int,

pysatl_experiment/experiment_new/step/execution/power/power.py

@@ -1,5 +1,7 @@
 from dataclasses import dataclass
 
+from line_profiler import profile
+
 from pysatl_criterion.statistics.goodness_of_fit import AbstractGoodnessOfFitStatistic
 from pysatl_experiment.configuration.model.alternative.alternative import Alternative
 from pysatl_experiment.experiment_new.step.execution.common.execution_step_data.execution_step_data import (
@@ -42,6 +44,7 @@ def __init__(
         self.result_storage = result_storage
         self.storage_connection = storage_connection
 
+    @profile
     def run(self) -> None:
         """
         Run standard power execution step.

pysatl_experiment/experiment_new/step/execution/time_complexity/time_complexity.py

@@ -1,5 +1,7 @@
 from dataclasses import dataclass
 
+from line_profiler import profile
+
 from pysatl_experiment.experiment_new.step.execution.common.execution_step_data.execution_step_data import (  # noqa: E501
     ExecutionStepData,
 )
@@ -43,6 +45,7 @@ def __init__(
         self.data_storage = data_storage
         self.result_storage = result_storage
 
+    @profile
     def run(self) -> None:
         """
         Run standard time complexity execution step.

pysatl_experiment/experiment_new/step/generation/generation.py

@@ -1,5 +1,7 @@
 from dataclasses import dataclass
 
+from line_profiler import profile
+
 from pysatl_experiment.experiment.generator import AbstractRVSGenerator
 from pysatl_experiment.persistence.model.random_values.random_values import IRandomValuesStorage, RandomValuesModel
 
@@ -30,6 +32,7 @@ def __init__(
         self.step_config = step_config
         self.data_storage = data_storage
 
+    @profile
     def run(self) -> None:
         """
         Run standard generation step.
@@ -39,6 +42,7 @@ def run(self) -> None:
             samples = self._generate_samples(step_data)
             self._save_samples_to_storage(samples, step_data.sample_size, step_data)
 
+    @profile
     def _generate_samples(self, step_data: GenerationStepData) -> list[list[float]]:
         """
         Generate samples.

pysatl_experiment/experiment_new/step/report_building/critical_value/critical_value.py

@@ -1,5 +1,7 @@
 from pathlib import Path
 
+from line_profiler import profile
+
 from pysatl_criterion.cv_calculator.cv_calculator.cv_calculator import CVCalculator
 from pysatl_criterion.persistence.model.limit_distribution.limit_distribution import (
     ILimitDistributionStorage,
@@ -33,6 +35,7 @@ def __init__(
         self.results_path = results_path
         self.with_chart = with_chart
 
+    @profile
     def run(self) -> None:
         """
         Run standard critical value report building step.

pysatl_experiment/experiment_new/step/report_building/power/power.py

@@ -1,5 +1,7 @@
 from pathlib import Path
 
+from line_profiler import profile
+
 from pysatl_experiment.configuration.criteria_config.criteria_config import CriterionConfig
 from pysatl_experiment.configuration.model.alternative.alternative import Alternative
 from pysatl_experiment.configuration.model.report_mode.report_mode import ReportMode
@@ -32,6 +34,7 @@ def __init__(
         self.results_path = results_path
         self.with_chart = with_chart
 
+    @profile
     def run(self) -> None:
         """
         Run standard power report building step.

pysatl_experiment/experiment_new/step/report_building/time_complexity/time_complexity.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 
 import numpy as np
+from line_profiler import profile
 
 from pysatl_experiment.configuration.criteria_config.criteria_config import CriterionConfig
 from pysatl_experiment.configuration.model.report_mode.report_mode import ReportMode
@@ -32,6 +33,7 @@ def __init__(
         self.results_path = results_path
         self.with_chart = with_chart
 
+    @profile
     def run(self) -> None:
         """
         Run standard time complexity report building step.