Skip to content
Snippets Groups Projects
Commit 9277ca55 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Add test for Tensorflow wrapping

parent f2abee0d
No related merge requests found
Pipeline #22303 failed with stage
in 2 minutes and 13 seconds
......@@ -24,4 +24,10 @@ def test_wrap_walberla():
def test_wrap_tensorflow():
pass
import pytest
pytest.importorskip("pystencils_autodiff")
from pystencils_autodiff.backends.astnodes import TensorflowModule
generate_shared_object(tempfile.TemporaryDirectory, None, show_code=True,
framework_module_class=TensorflowModule, generate_code_only=True)
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment