Avoid zero-tensor warning

lab-cosmo · Feb 7, 2024 · 21ca03c · 21ca03c
1 parent aa4b789
commit 21ca03c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_spex/radial_basis.py b/torch_spex/radial_basis.py
@@ -58,7 +58,7 @@ def __init__(self, hypers, all_species,
         else:
             self.is_alchemical = False
             self.n_pseudo_species = 0  # dummy for torchscript
-            self.combination_matrix = torch.nn.Linear(0, 0)  # dummy for torchscript
+            self.combination_matrix = torch.nn.Linear(1, 1)  # dummy for torchscript
             self.species_neighbor_labels = Labels.empty("dummy")
 
         self.apply_mlp = False