Lint

src/pystencils_autodiff/backends/_torch_native.py

@@ -106,8 +106,7 @@ def generate_torch(destination_folder,
         }
 
         if is_cuda:
-            template_string_cpp = read_file(join(dirname(__file__),
-                                                  'torch_native_cuda.tmpl.cpp'))
+            template_string_cpp = read_file(join(dirname(__file__), 'torch_native_cuda.tmpl.cpp'))
             template = jinja2.Template(template_string_cpp)
             output = template.render(render_dict)
             write_file(join(destination_folder, operation_string + '.cpp'), output)
@@ -117,8 +116,7 @@ def generate_torch(destination_folder,
             output = template.render(render_dict)
             write_file(join(destination_folder, operation_string + '.cu'), output)
         else:
-            template_string_cpp = read_file(join(dirname(__file__),
-                                                  'torch_native_cpu.tmpl.cpp'))
+            template_string_cpp = read_file(join(dirname(__file__), 'torch_native_cpu.tmpl.cpp'))
             template = jinja2.Template(template_string_cpp)
             output = template.render(render_dict)
             write_file(join(destination_folder, operation_string + '.cpp'), output)

src/pystencils_autodiff/backends/python_bindings.py

@@ -12,7 +12,7 @@ import itertools
 import jinja2
 import stringcase
 
-from pystencils.astnodes import KernelFunction, Node
+from pystencils.astnodes import KernelFunction
 from pystencils_autodiff.framework_integration.astnodes import JinjaCppFile
 
 

src/pystencils_autodiff/tensorflow_jit.py

@@ -10,7 +10,6 @@
 import json
 import subprocess
 import sysconfig
-from itertools import chain
 from os.path import exists, join
 
 from tqdm import tqdm