v0.1.1

mbarbetti · Nov 10, 2023 · 37d96dc · 37d96dc
1 parent 0578b0b
commit 37d96dc
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 6 deletions.
diff --git a/src/pidgan/players/generators/Generator.py b/src/pidgan/players/generators/Generator.py
@@ -2,7 +2,6 @@
 from tensorflow import keras
 
 LEAKY_ALPHA = 0.1
-SEED = 42
 
 
 class Generator(keras.Model):
@@ -106,7 +105,7 @@ def summary(self, **kwargs) -> None:
         self._seq.summary(**kwargs)
 
     def generate(self, x, seed=None, return_latent_sample=False) -> tf.Tensor:
-        tf.random.set_seed(seed=SEED)
+        tf.random.set_seed(seed=seed)
         x, latent_sample = self._prepare_input(x, seed=seed)
         out = self._seq(x)
         if return_latent_sample:

diff --git a/src/pidgan/version.py b/src/pidgan/version.py
@@ -1 +1 @@
-__version__ = "0.1.0"
+__version__ = "0.1.1"
diff --git a/tests/players/generators/test_Generator.py b/tests/players/generators/test_Generator.py
@@ -94,10 +94,13 @@ def test_model_eval(model, sample_weight):
 
 
 def test_model_generate(model):
-    out = model.generate(x, seed=42)
-    comparison = out.numpy() == model.generate(x, seed=42).numpy()
+    no_seed_out = model.generate(x, seed=None)
+    comparison = no_seed_out.numpy() != model.generate(x, seed=None).numpy()
     assert comparison.all()
-    comparison = out.numpy() != model.generate(x, seed=24).numpy()
+    seed_out = model.generate(x, seed=42)
+    comparison = seed_out.numpy() == model.generate(x, seed=42).numpy()
+    assert comparison.all()
+    comparison = seed_out.numpy() != model.generate(x, seed=24).numpy()
     assert comparison.any()