Merge pull request #4 from idiap/keyboard-interrupt

Return correct error code 130 on keyboard interrupt
idiap · May 3, 2024 · 159f90a · 159f90a
2 parents 5b32333 + 2624e31
commit 159f90a
Show file tree

Hide file tree

Showing 5 changed files with 92 additions and 113 deletions.
diff --git a/requirements.dev.txt b/requirements.dev.txt
@@ -1,4 +1,4 @@
-black
+black==24.2.0
 coverage
 isort
 pytest

diff --git a/tests/test_train_gan.py b/tests/test_train_gan.py
@@ -79,8 +79,7 @@ def __init__(self):
             self.generator = Generator(latent_dim=100, img_shape=data_shape)
             self.discriminator = Discriminator(img_shape=data_shape)
 
-        def forward(self, x):
-            ...
+        def forward(self, x): ...
 
         def train_step(self, batch, criterion, optimizer_idx):
             imgs, _ = batch
@@ -174,8 +173,7 @@ def __init__(self):
             self.generator = Generator(latent_dim=100, img_shape=data_shape)
             self.discriminator = Discriminator(img_shape=data_shape)
 
-        def forward(self, x):
-            ...
+        def forward(self, x): ...
 
         def train_step(self, batch, criterion, optimizer_idx):
             imgs, _ = batch
@@ -271,8 +269,7 @@ def __init__(self):
             self.generator = Generator(latent_dim=100, img_shape=data_shape)
             self.discriminator = Discriminator(img_shape=data_shape)
 
-        def forward(self, x):
-            ...
+        def forward(self, x): ...
 
         def optimize(self, batch, trainer):
             imgs, _ = batch
@@ -385,8 +382,7 @@ def __init__(self):
             self.generator = Generator(latent_dim=100, img_shape=data_shape)
             self.discriminator = Discriminator(img_shape=data_shape)
 
-        def forward(self, x):
-            ...
+        def forward(self, x): ...
 
         def optimize(self, batch, trainer):
             imgs, _ = batch
@@ -502,11 +498,9 @@ def __init__(self):
             self.generator = Generator(latent_dim=100, img_shape=data_shape)
             self.discriminator = Discriminator(img_shape=data_shape)
 
-        def train_step():
-            ...
+        def train_step(): ...
 
-        def forward(self, x):
-            ...
+        def forward(self, x): ...
 
         def optimize(self, batch, trainer):
             imgs, _ = batch

diff --git a/trainer/VERSION b/trainer/VERSION
@@ -1 +1 @@
-v0.1.0
+v0.1.1