From 0288d0aa58bdfd17b5d33339d930390145532955 Mon Sep 17 00:00:00 2001 From: nbouziani Date: Tue, 7 Feb 2023 13:15:15 +0000 Subject: [PATCH] Rename torch_op -> torch_operator --- firedrake/external_operators/__init__.py | 1 + firedrake/external_operators/neural_networks/__init__.py | 2 ++ .../external_operators/neural_networks/ml_backend_coupling.py | 2 +- 3 files changed, 4 insertions(+), 1 deletion(-) create mode 100644 firedrake/external_operators/__init__.py create mode 100644 firedrake/external_operators/neural_networks/__init__.py diff --git a/firedrake/external_operators/__init__.py b/firedrake/external_operators/__init__.py new file mode 100644 index 0000000000..d37df34173 --- /dev/null +++ b/firedrake/external_operators/__init__.py @@ -0,0 +1 @@ +from firedrake.external_operators.neural_networks import * \ No newline at end of file diff --git a/firedrake/external_operators/neural_networks/__init__.py b/firedrake/external_operators/neural_networks/__init__.py new file mode 100644 index 0000000000..6be941f2dd --- /dev/null +++ b/firedrake/external_operators/neural_networks/__init__.py @@ -0,0 +1,2 @@ +from .backends import get_backend +from .ml_backend_coupling import HybridOperator, torch_operator \ No newline at end of file diff --git a/firedrake/external_operators/neural_networks/ml_backend_coupling.py b/firedrake/external_operators/neural_networks/ml_backend_coupling.py index c1c5463d13..bc5fff7b28 100644 --- a/firedrake/external_operators/neural_networks/ml_backend_coupling.py +++ b/firedrake/external_operators/neural_networks/ml_backend_coupling.py @@ -31,5 +31,5 @@ def __call__(self, *ω): return φ(*ω) -def torch_op(*args, **kwargs): +def torch_operator(*args, **kwargs): return HybridOperator(*args, **kwargs)