update

pyg-team · Jul 26, 2022 · c4507eb · c4507eb
1 parent 5b4027e
commit c4507eb
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torch_geometric/loader/dataloader.py b/torch_geometric/loader/dataloader.py
@@ -56,8 +56,7 @@ def pin_memory(data, device=None):
         return data.pin_memory()
     if len(__torch_pin_memory_params) > 1:
         return __torch_pin_memory(data, device)
-    else:
-        return __torch_pin_memory(data)
+    return __torch_pin_memory(data)
 
 
 torch.utils.data._utils.pin_memory.pin_memory = pin_memory