Run Black (#106)

Co-authored-by: knikolaou <>
zincware · Nov 3, 2023 · dc5f814 · dc5f814
1 parent 9e68f62
commit dc5f814
Show file tree

Hide file tree

Showing 105 changed files with 103 additions and 2 deletions.
diff --git a/CI/integration_tests/agent/test_approximate_max_entropy.py b/CI/integration_tests/agent/test_approximate_max_entropy.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import optax
 from neural_tangents import stax
 

diff --git a/CI/integration_tests/training_recording/test_training_recording_deployment.py b/CI/integration_tests/training_recording/test_training_recording_deployment.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/integration_tests/training_strategies/test_loss_aware_reservoir_deployment.py b/CI/integration_tests/training_strategies/test_loss_aware_reservoir_deployment.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/integration_tests/training_strategies/test_partitioned_training_deployment.py b/CI/integration_tests/training_strategies/test_partitioned_training_deployment.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/analysis/test_eigen_space_analysis.py b/CI/unit_tests/analysis/test_eigen_space_analysis.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import jax.random as random
 from jax.lib import xla_bridge
 

diff --git a/CI/unit_tests/analysis/test_entropy_anaysis.py b/CI/unit_tests/analysis/test_entropy_anaysis.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/analysis/test_ntk_observables.py b/CI/unit_tests/analysis/test_ntk_observables.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/data/test_abalone.py b/CI/unit_tests/data/test_abalone.py
@@ -25,6 +25,7 @@
 -------
 Test Abaone generator.
 """
+
 from znnl.data import AbaloneDataGenerator
 
 

diff --git a/CI/unit_tests/data/test_cifar.py b/CI/unit_tests/data/test_cifar.py
@@ -25,6 +25,7 @@
 -------
 Test CIFAR10 generator.
 """
+
 from znnl.data import CIFAR10Generator
 
 

diff --git a/CI/unit_tests/data/test_confined_particles.py b/CI/unit_tests/data/test_confined_particles.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/data/test_data_generator.py b/CI/unit_tests/data/test_data_generator.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/data/test_decision_boundary.py b/CI/unit_tests/data/test_decision_boundary.py
@@ -25,6 +25,7 @@
 -------
 Unit test for the decision boundary.
 """
+
 import jax.numpy as np
 import numpy as onp
 from pytest import approx

diff --git a/CI/unit_tests/data/test_mnist.py b/CI/unit_tests/data/test_mnist.py
@@ -25,6 +25,7 @@
 -------
 Test MNIST generator.
 """
+
 from znnl.data import MNISTGenerator
 
 

diff --git a/CI/unit_tests/data/test_mpg.py b/CI/unit_tests/data/test_mpg.py
@@ -25,6 +25,7 @@
 -------
 Test MPG generator.
 """
+
 from znnl.data import MPGDataGenerator
 
 

diff --git a/CI/unit_tests/data/test_points_on_circle.py b/CI/unit_tests/data/test_points_on_circle.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/data/test_points_on_lattice.py b/CI/unit_tests/data/test_points_on_lattice.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/distance_metrics/test_angular_distance.py b/CI/unit_tests/distance_metrics/test_angular_distance.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/distance_metrics/test_cosine_distance.py b/CI/unit_tests/distance_metrics/test_cosine_distance.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/distance_metrics/test_hyper_sphere_distance.py b/CI/unit_tests/distance_metrics/test_hyper_sphere_distance.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/distance_metrics/test_l_p_norm.py b/CI/unit_tests/distance_metrics/test_l_p_norm.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/distance_metrics/test_mahalanobis_distance.py b/CI/unit_tests/distance_metrics/test_mahalanobis_distance.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/distance_metrics/test_order_n_difference.py b/CI/unit_tests/distance_metrics/test_order_n_difference.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/loss_functions/test_loss_functions.py b/CI/unit_tests/loss_functions/test_loss_functions.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/models/test_flax_model.py b/CI/unit_tests/models/test_flax_model.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/models/test_nt_model.py b/CI/unit_tests/models/test_nt_model.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/models/test_seed.py b/CI/unit_tests/models/test_seed.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/optimizers/test_trace_optimizer.py b/CI/unit_tests/optimizers/test_trace_optimizer.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/point_selection/test_greedy_selection.py b/CI/unit_tests/point_selection/test_greedy_selection.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/similarity_measures/test_cosine_similarity.py b/CI/unit_tests/similarity_measures/test_cosine_similarity.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/training_recording/test_data_storage.py b/CI/unit_tests/training_recording/test_data_storage.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import tempfile
 from dataclasses import dataclass
 from os import path

diff --git a/CI/unit_tests/training_recording/test_training_recording.py b/CI/unit_tests/training_recording/test_training_recording.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import tempfile
 from pathlib import Path
 

diff --git a/CI/unit_tests/training_strategies/test_loss_aware_reservoir.py b/CI/unit_tests/training_strategies/test_loss_aware_reservoir.py
@@ -25,6 +25,7 @@
 -------
 Unit tests for the loss aware reservoir training class.
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/training_strategies/test_partitioned_training.py b/CI/unit_tests/training_strategies/test_partitioned_training.py
@@ -25,6 +25,7 @@
 -------
 Unit tests for the partitioned training class.
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/training_strategies/test_simple_training.py b/CI/unit_tests/training_strategies/test_simple_training.py
@@ -25,6 +25,7 @@
 -------
 Test the RND class.
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/utils/test_matrix_utils.py b/CI/unit_tests/utils/test_matrix_utils.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 
 os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

diff --git a/CI/unit_tests/utils/test_prng.py b/CI/unit_tests/utils/test_prng.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 from itertools import combinations
 
 import jax.random

diff --git a/CI/unit_tests/visualization/test_tsne_visualizer.py b/CI/unit_tests/visualization/test_tsne_visualizer.py
@@ -25,7 +25,6 @@
 -------
 """
 
-
 class TestTSNEVisualizer:
     """
     Test for the tSNE visualizer.

diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import os
 import sys
 

diff --git a/setup.py b/setup.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 from os import path
 
 import setuptools

diff --git a/znnl/__init__.py b/znnl/__init__.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import logging
 
 import znnl

diff --git a/znnl/accuracy_functions/__init__.py b/znnl/accuracy_functions/__init__.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 from znnl.accuracy_functions.accuracy_function import AccuracyFunction
 from znnl.accuracy_functions.label_accuracy import LabelAccuracy
 

diff --git a/znnl/accuracy_functions/accuracy_function.py b/znnl/accuracy_functions/accuracy_function.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import jax.numpy as np
 
 

diff --git a/znnl/accuracy_functions/label_accuracy.py b/znnl/accuracy_functions/label_accuracy.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import jax.numpy as np
 
 from znnl.accuracy_functions.accuracy_function import AccuracyFunction

diff --git a/znnl/agents/__init__.py b/znnl/agents/__init__.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 from znnl.agents.agent import Agent
 from znnl.agents.approximate_maximum_entropy import ApproximateMaximumEntropy
 from znnl.agents.maximum_entropy import MaximumEntropy

diff --git a/znnl/agents/agent.py b/znnl/agents/agent.py
@@ -25,7 +25,6 @@
 -------
 """
 
-
 class Agent:
     """
     Parent class of the data selection agents.

diff --git a/znnl/agents/approximate_maximum_entropy.py b/znnl/agents/approximate_maximum_entropy.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import jax
 import jax.numpy as np
 

diff --git a/znnl/agents/maximum_entropy.py b/znnl/agents/maximum_entropy.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 from znnl.agents.agent import Agent
 
 

diff --git a/znnl/agents/random.py b/znnl/agents/random.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import jax
 import jax.numpy as np
 

diff --git a/znnl/agents/rnd.py b/znnl/agents/rnd.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 import copy
 import logging
 import time

diff --git a/znnl/analysis/__init__.py b/znnl/analysis/__init__.py
@@ -24,6 +24,7 @@
 Summary
 -------
 """
+
 from znnl.analysis.eigensystem import EigenSpaceAnalysis
 from znnl.analysis.entropy import EntropyAnalysis
 from znnl.analysis.loss_fn_derivative import LossDerivative