From 0cccbe5eafc9f8e99af9e0f59fb7b65254c0d214 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 12 Sep 2019 18:55:43 +0200
Subject: [PATCH] Lint

---
 src/pystencils_autodiff/backends/_torch_native.py   | 6 ++----
 src/pystencils_autodiff/backends/python_bindings.py | 2 +-
 src/pystencils_autodiff/tensorflow_jit.py           | 1 -
 3 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/src/pystencils_autodiff/backends/_torch_native.py b/src/pystencils_autodiff/backends/_torch_native.py
index d63db1d..53619b5 100644
--- a/src/pystencils_autodiff/backends/_torch_native.py
+++ b/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)
diff --git a/src/pystencils_autodiff/backends/python_bindings.py b/src/pystencils_autodiff/backends/python_bindings.py
index 3cae1a8..d7acce1 100644
--- a/src/pystencils_autodiff/backends/python_bindings.py
+++ b/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
 
 
diff --git a/src/pystencils_autodiff/tensorflow_jit.py b/src/pystencils_autodiff/tensorflow_jit.py
index 76e887c..be3fbf9 100644
--- a/src/pystencils_autodiff/tensorflow_jit.py
+++ b/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
-- 
GitLab