Spaces:
Sleeping
Sleeping
MilesCranmer
commited on
Commit
•
215a7a1
1
Parent(s):
44dcbea
More comments for torch test
Browse files- test/test_torch.py +2 -1
test/test_torch.py
CHANGED
@@ -7,12 +7,13 @@ from pysr import sympy2torch, PySRRegressor
|
|
7 |
import platform
|
8 |
|
9 |
if platform.system() == "Darwin":
|
|
|
10 |
from pysr.julia_helpers import init_julia
|
11 |
|
12 |
Main = init_julia()
|
13 |
import torch
|
14 |
else:
|
15 |
-
#
|
16 |
# https://github.com/pytorch/pytorch/issues/78829
|
17 |
import torch
|
18 |
import sympy
|
|
|
7 |
import platform
|
8 |
|
9 |
if platform.system() == "Darwin":
|
10 |
+
# Import PyJulia, then Torch
|
11 |
from pysr.julia_helpers import init_julia
|
12 |
|
13 |
Main = init_julia()
|
14 |
import torch
|
15 |
else:
|
16 |
+
# Import Torch, then PyJulia
|
17 |
# https://github.com/pytorch/pytorch/issues/78829
|
18 |
import torch
|
19 |
import sympy
|