Skip to content

Commit

Permalink
Rename torch_op -> torch_operator
Browse files Browse the repository at this point in the history
  • Loading branch information
nbouziani committed Feb 7, 2023
1 parent 422b1d5 commit 0288d0a
Showing 3 changed files with 4 additions and 1 deletion.
1 change: 1 addition & 0 deletions firedrake/external_operators/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
from firedrake.external_operators.neural_networks import *
2 changes: 2 additions & 0 deletions firedrake/external_operators/neural_networks/__init__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
from .backends import get_backend
from .ml_backend_coupling import HybridOperator, torch_operator
Original file line number Diff line number Diff line change
@@ -31,5 +31,5 @@ def __call__(self, *ω):
return φ(*ω)


def torch_op(*args, **kwargs):
def torch_operator(*args, **kwargs):
return HybridOperator(*args, **kwargs)

0 comments on commit 0288d0a

Please sign in to comment.