From a1039010fa15ab5a6662094113606f89f3e73c9a Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 5 Nov 2019 18:39:43 +0100
Subject: [PATCH] Lint
---
src/pystencils_autodiff/backends/astnodes.py | 2 +-
src/pystencils_autodiff/field_tensor_conversion.py | 5 -----
2 files changed, 1 insertion(+), 6 deletions(-)
diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py
index 237721b..7a7686a 100644
--- a/src/pystencils_autodiff/backends/astnodes.py
+++ b/src/pystencils_autodiff/backends/astnodes.py
@@ -169,7 +169,7 @@ setup_pybind11(cfg)
source_code = self.CPP_IMPORT_PREFIX + str(self)
hash_str = _hash(source_code.encode()).hexdigest()
cache_dir = join(get_cache_config()['object_cache'], f'cppimport_{hash_str}')
- file_name = join(cache_dir, f'{self.module_name}.cpp')
+ file_name = join(cache_dir, f'{self.module_name}.cpp')
os.makedirs(cache_dir, exist_ok=True)
if not exists(file_name):
diff --git a/src/pystencils_autodiff/field_tensor_conversion.py b/src/pystencils_autodiff/field_tensor_conversion.py
index 72382b1..7c14294 100644
--- a/src/pystencils_autodiff/field_tensor_conversion.py
+++ b/src/pystencils_autodiff/field_tensor_conversion.py
@@ -48,11 +48,6 @@ def create_field_from_array_like(field_name, maybe_array, annotations=None):
index_dimensions = 0
field_type = FieldType.GENERIC
- try:
- import torch
- except ImportError:
- torch = None
-
if 'tensorflow.python.' in str(type(maybe_array)) and 'Tensor' in str(type(maybe_array)):
try:
# This fails on eager execution
--
GitLab