fix: squeeze (#33)

* fix: squeeze

* chore: bump version

* Downgrade version from 1.0.2 to 1.0.1

---------

Co-authored-by: Jannis Born <jannis.born@gmx.de>

Author: MAlberts99

paccmann_predictor/utils/utils.py

@@ -48,7 +48,7 @@ class Squeeze(nn.Module):
     """Squeeze wrapper for nn.Sequential."""
 
     def forward(self, data):
-        return torch.squeeze(data)
+        return torch.squeeze(data, -1)
 
 
 class Unsqueeze(nn.Module):