fix config store

pyg-team · Sep 13, 2022 · fe33faa · fe33faa
1 parent cdb06a5
commit fe33faa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_geometric/graphgym/config_store.py b/torch_geometric/graphgym/config_store.py
@@ -11,6 +11,7 @@
 
 MAPPING = {
     torch.nn.Module: Any,
+    torch.Tensor: Any,
 }