Spaces:
Sleeping
Sleeping
MilesCranmer
commited on
Commit
•
9b2a102
1
Parent(s):
5ffac80
Fix formatting
Browse files- test/test_torch.py +3 -0
test/test_torch.py
CHANGED
@@ -2,10 +2,13 @@ import unittest
|
|
2 |
import numpy as np
|
3 |
import pandas as pd
|
4 |
from pysr import sympy2torch, PySRRegressor
|
|
|
5 |
# Need to initialize Julia before importing torch...
|
6 |
import platform
|
|
|
7 |
if platform.system() == "Darwin":
|
8 |
from pysr.julia_helpers import init_julia
|
|
|
9 |
Main = init_julia()
|
10 |
import torch
|
11 |
else:
|
|
|
2 |
import numpy as np
|
3 |
import pandas as pd
|
4 |
from pysr import sympy2torch, PySRRegressor
|
5 |
+
|
6 |
# Need to initialize Julia before importing torch...
|
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:
|