From 2f3b5b1f9538d41f47ed362ca0343d04f3824af6 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 7 Aug 2019 19:38:22 +0200
Subject: [PATCH] Lint
---
src/pystencils_autodiff/backends/_pytorch.py | 4 +++-
src/pystencils_autodiff/backends/_tensorflow_cpp.py | 3 ++-
src/pystencils_autodiff/lbm/__init__.py | 2 ++
src/pystencils_autodiff/tensorflow.py | 2 --
4 files changed, 7 insertions(+), 4 deletions(-)
diff --git a/src/pystencils_autodiff/backends/_pytorch.py b/src/pystencils_autodiff/backends/_pytorch.py
index 6d2da60..2224276 100644
--- a/src/pystencils_autodiff/backends/_pytorch.py
+++ b/src/pystencils_autodiff/backends/_pytorch.py
@@ -100,7 +100,9 @@ def tensor_to_gpuarray(tensor):
raise ValueError(
'Cannot convert CPU tensor to GPUArray (call `cuda()` on it)')
else:
- return pycuda.gpuarray.GPUArray(tensor.shape, dtype=torch_dtype_to_numpy(tensor.dtype), gpudata=tensor.data_ptr())
+ return pycuda.gpuarray.GPUArray(tensor.shape,
+ dtype=torch_dtype_to_numpy(tensor.dtype),
+ gpudata=tensor.data_ptr())
def gpuarray_to_tensor(gpuarray, context=None):
diff --git a/src/pystencils_autodiff/backends/_tensorflow_cpp.py b/src/pystencils_autodiff/backends/_tensorflow_cpp.py
index 541e122..d64f176 100644
--- a/src/pystencils_autodiff/backends/_tensorflow_cpp.py
+++ b/src/pystencils_autodiff/backends/_tensorflow_cpp.py
@@ -6,7 +6,8 @@ Advantages:
- Custom Ops will be serializable
Disadavantages:
-- C++ Code has to be build with correct parameters and ABI for present Tensorflow version (best integrated into Tensorflow build)
+- C++ Code has to be build with correct parameters and ABI
+for present Tensorflow version (best integrated into Tensorflow build)
"""
diff --git a/src/pystencils_autodiff/lbm/__init__.py b/src/pystencils_autodiff/lbm/__init__.py
index b057947..a28beeb 100644
--- a/src/pystencils_autodiff/lbm/__init__.py
+++ b/src/pystencils_autodiff/lbm/__init__.py
@@ -1,2 +1,4 @@
from lbmpy.autodiff._autodiff_lbstep import AutoDiffLatticeBoltzmannStep
from lbmpy.autodiff.adjoint_boundaryconditions import AdjointBoundaryCondition
+
+__all__ = ['AutoDiffLatticeBoltzmannStep', 'AdjointBoundaryCondition']
diff --git a/src/pystencils_autodiff/tensorflow.py b/src/pystencils_autodiff/tensorflow.py
index 82a86cc..91a8c76 100644
--- a/src/pystencils_autodiff/tensorflow.py
+++ b/src/pystencils_autodiff/tensorflow.py
@@ -24,5 +24,3 @@ def tf_variable_from_field(field, init_val=0, constraint=None):
def tf_placeholder_from_field(field):
return tf.placeholder(dtype=field.dtype.numpy_dtype, name=field.name + '_placeholder', shape=field.shape)
-
-
--
GitLab