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