Spaces:
Sleeping
Sleeping
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:
|