Update test.py

JasonKaV · Feb 9, 2024 · 5066c9c · 5066c9c
1 parent 8ba868b
commit 5066c9c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/OpenDFT/QHBench/QH9/test.py b/OpenDFT/QHBench/QH9/test.py
@@ -218,7 +218,7 @@ def main(conf):
 
 
 def post_processing(batch, default_type):
-    for key in batch.keys:
+    for key in batch.keys():
         if torch.is_floating_point(batch[key]):
             batch[key] = batch[key].type(default_type)
     return batch