Add/update/fix tests

braun-steven · Dec 5, 2023 · 6912a9c · 6912a9c
1 parent 051e606
commit 6912a9c
Show file tree

Hide file tree

Showing 9 changed files with 1 addition and 313 deletions.
diff --git a/simple_einet/sampling_utils.py b/simple_einet/sampling_utils.py
@@ -303,6 +303,7 @@ def init_einet_stats(einet: "Einet", dataloader: torch.utils.data.DataLoader):
 
     # Compute mean and std
     from tqdm import tqdm
+
     for batch in tqdm(dataloader, desc="Leaf Parameter Initialization"):
         data, label = batch
         if stats_mean == None:

diff --git a/simple_einet/tests/__init__.py b/simple_einet/tests/__init__.py
diff --git a/simple_einet/tests/layers/__init__.py b/simple_einet/tests/layers/__init__.py
diff --git a/simple_einet/tests/layers/distributions/__init__.py b/simple_einet/tests/layers/distributions/__init__.py
diff --git a/simple_einet/tests/layers/test_einsum_layer.py b/simple_einet/tests/layers/test_einsum_layer.py
diff --git a/simple_einet/tests/layers/test_linsum_layer.py b/simple_einet/tests/layers/test_linsum_layer.py
diff --git a/simple_einet/tests/layers/test_mixing_layer.py b/simple_einet/tests/layers/test_mixing_layer.py
diff --git a/simple_einet/tests/layers/test_sum_layer.py b/simple_einet/tests/layers/test_sum_layer.py
diff --git a/simple_einet/tests/layers/test_utils.py b/simple_einet/tests/layers/test_utils.py