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
Branches
Tags
No related merge requests found
Pipeline #22303 failed
...@@ -24,4 +24,10 @@ def test_wrap_walberla(): ...@@ -24,4 +24,10 @@ def test_wrap_walberla():
def test_wrap_tensorflow(): 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% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment