Spaces:

MilesCranmer
/

PySR

Running

MilesCranmer commited on Jun 13, 2021

Commit

4d5aec3

1 Parent(s): 10646c2

Correct sign in torch modulo

Files changed (1) hide show

pysr/export_torch.py CHANGED Viewed

@@ -76,7 +76,7 @@ def _initialize_torch():
             sympy.Not: torch.logical_not,
             sympy.Max: torch.max,
             sympy.Min: torch.min,
-            sympy.Mod: torch.fmod,
         }
         class _Node(torch.nn.Module):

             sympy.Not: torch.logical_not,
             sympy.Max: torch.max,
             sympy.Min: torch.min,
+            sympy.Mod: torch.remainder,
         }
         class _Node(torch.nn.Module):