fixed small errors in sample notebooks

Author: facundoq

docs/examples/ResNet Invariance with TinyImageNet.ipynb

@@ -35,6 +35,8 @@ dev = [
     "poethepoet>=0.35.0",
     "data-science-types",
     "poutyne",
+    "ipython>=9.0.2",
+    "tinyimagenet>=0.9.9",
 ]
 docs = [
     "sphinx>=8.2.3",
@@ -137,4 +139,4 @@ format = "uvx ruff format tmeasures"
 build = "uv build"
 docs = "uv run sphinx-build docs docs/_build"
 serve-docs = " uv run sphinx-autobuild --open-browser docs docs/_build"
-precommit = "uv run pre-commit run --all-files"
+precommit = "uv run pre-commit run --all-files"

docs/examples/ResNet_rotation_invariance.pkl

@@ -1,4 +1,4 @@
-from tmeasures.pytorch.stats_running import RunningMeanAndVarianceWelford, RunningMeanWelford,RunningMean,RunningVariance,RunningMeanSets,RunningMeanVarianceSets
+from tmeasures.pytorch.measures.stats_running import RunningMeanAndVarianceWelford, RunningMeanWelford,RunningMean,RunningVariance,RunningMeanSets,RunningMeanVarianceSets
 import itertools
 import torch
 import pytest

docs/examples/Variance to rotations of a CNN trained on MNIST with PyTorch.ipynb

@@ -30,11 +30,10 @@
 from .by_layer import PyTorchMeasureByLayer
 
 from .measure_transformer import MeasureTransformation,NoTransformation,AverageFeatureMaps
-from .util import SequentialWithIntermediates
 from . import dataset2d
 
 from .layer_measures import Variance
 
-from .variance_invariance import SampleVarianceInvariance,TransformationVarianceInvariance,NormalizedVarianceInvariance
-from .variance_sameequivariance import  SampleVarianceSameEquivariance,TransformationVarianceSameEquivariance, NormalizedVarianceSameEquivariance
-from .goodfellow import GoodfellowInvariance,NormalPValueThreshold,PercentActivationThreshold
+from .measures.variance_invariance import SampleVarianceInvariance,TransformationVarianceInvariance,NormalizedVarianceInvariance
+from .measures.variance_sameequivariance import  SampleVarianceSameEquivariance,TransformationVarianceSameEquivariance, NormalizedVarianceSameEquivariance
+from .measures.goodfellow import GoodfellowInvariance,NormalPValueThreshold,PercentActivationThreshold

docs/examples/_layer3_BasicBlock_1_conv2_rgb_.png

@@ -1,6 +1,6 @@
 from .. import logger as tm_logger
 from .base import PyTorchActivationMeasure, STMatrixIterator
-from .stats_running import RunningMeanAndVarianceWelford, RunningMeanWelford
+from .measures.stats_running import RunningMeanAndVarianceWelford, RunningMeanWelford
 
 logger = tm_logger.getChild("pytorch.layer_measures")
 

pyproject.toml

@@ -7,7 +7,7 @@
 import tmeasures as tm
 from tmeasures.pytorch.transformations import IdentityTransformationSet
 
-from . import (
+from .. import (
     BaseActivationsModule,
     PyTorchActivationMeasure,
     PyTorchMeasure,
@@ -16,8 +16,8 @@
     PyTorchMeasureResult,
     STMatrixIterator,
 )
-from .activations_iterator import PytorchActivationsIterator
-from .layer_measures import Variance
+from ..activations_iterator import PytorchActivationsIterator
+from ..layer_measures import Variance
 from .quotient import divide_activations
 from .stats_running import RunningMeanAndVarianceWelford, RunningMeanWelford
 

tests/pytorch/test_stats_running.py

@@ -2,9 +2,9 @@
 import tqdm.auto as tqdm
 from torch.utils.data import Dataset
 
-from .. import Measure, MeasureResult, TransformationSet
-from . import ActivationsByLayer, BaseActivationsModule, PyTorchMeasure, PyTorchMeasureOptions, PyTorchMeasureResult
-from .measure_transformer import MeasureTransformation, NoTransformation
+from ... import Measure, MeasureResult, TransformationSet
+from .. import ActivationsByLayer, BaseActivationsModule, PyTorchMeasure, PyTorchMeasureOptions, PyTorchMeasureResult
+from ..measure_transformer import MeasureTransformation, NoTransformation
 
 
 def safe_divide(x: torch.Tensor, y: torch.Tensor):