[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

pvnet/models/multimodal/site_encoders/encoders.py

@@ -74,24 +74,24 @@ def __init__(
         )
 
     def _calculate_attention(self, x):
-        i=0
+        i = 0
         for key in x.keys():
-            if(i==17):
-                ans=key
+            if i == 17:
+                ans = key
             print(key, type(key))
-            i+=1
+            i += 1
         gsp_ids = x[ans].squeeze().int()
         attention = self._attention_network(gsp_ids)
         return attention
 
     def _encode_value(self, x):
         # Shape: [batch size, sequence length, PV site]
-        i=0
+        i = 0
         for key in x.keys():
-            if(i==6):
-                ans=key
+            if i == 6:
+                ans = key
             print(key, type(key))
-            i+=1
+            i += 1
         pv_site_seqs = x[ans].float()
         batch_size = pv_site_seqs.shape[0]