MilesCranmer commited on
Commit
215a7a1
1 Parent(s): 44dcbea

More comments for torch test

Browse files
Files changed (1) hide show
  1. 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
- # Switch order of imports.
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