MilesCranmer commited on
Commit
aefe008
1 Parent(s): 87f1eda

docs: clearer about torch port

Browse files
Files changed (1) hide show
  1. pysr/export_torch.py +2 -2
pysr/export_torch.py CHANGED
@@ -82,7 +82,7 @@ def _initialize_torch():
82
  }
83
 
84
  class _Node(torch.nn.Module):
85
- """SympyTorch code from https://github.com/patrick-kidger/sympytorch"""
86
 
87
  def __init__(self, *, expr, _memodict, _func_lookup, **kwargs):
88
  super().__init__(**kwargs)
@@ -154,7 +154,7 @@ def _initialize_torch():
154
  return self._torch_func(*args)
155
 
156
  class _SingleSymPyModule(torch.nn.Module):
157
- """SympyTorch code from https://github.com/patrick-kidger/sympytorch"""
158
 
159
  def __init__(
160
  self, expression, symbols_in, selection=None, extra_funcs=None, **kwargs
 
82
  }
83
 
84
  class _Node(torch.nn.Module):
85
+ """Forked from https://github.com/patrick-kidger/sympytorch"""
86
 
87
  def __init__(self, *, expr, _memodict, _func_lookup, **kwargs):
88
  super().__init__(**kwargs)
 
154
  return self._torch_func(*args)
155
 
156
  class _SingleSymPyModule(torch.nn.Module):
157
+ """Forked from https://github.com/patrick-kidger/sympytorch"""
158
 
159
  def __init__(
160
  self, expression, symbols_in, selection=None, extra_funcs=None, **kwargs