diff --git a/environment.yml b/environment.yml
index 2869b2519a5bf1780f9567281c70eeaf1b5bc7d9..bd0b51da2dd66cae6d58f4ad5a129f2053c3689f 100644
--- a/environment.yml
+++ b/environment.yml
@@ -85,5 +85,4 @@ dependencies:
     - tzdata==2023.3
     - urllib3==2.0.2
     - xxhash==3.2.0
-    - yarl==1.9.2
-prefix: /home/lena/anaconda3/envs/testenv
+    - yarl==1.9.2
\ No newline at end of file
diff --git a/pytorch_classifier.py b/pytorch_classifier.py
index 0fcc27acdf4d18b35ff6e413505309dcd6f0634d..ad32fc783ed88c7133f7ff3a7cda56a6a02e0e3c 100644
--- a/pytorch_classifier.py
+++ b/pytorch_classifier.py
@@ -194,7 +194,6 @@ def train(model,
                 optimizer.step() # or flip them?
                 optimizer.zero_grad()
                 
-            
                 model.zero_grad()
 #             loss.backward()
                 torch.nn.utils.clip_grad_norm_(model.parameters(), 1.0)