diff --git a/src/pystencils_autodiff/backends/astnodes.py b/src/pystencils_autodiff/backends/astnodes.py index 237721b2a836d585cb25548809c4d1e5517be8d2..7a7686a1c3e384731e3b9b5ff16401012837eefb 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 72382b11ab592ce0b03df8d9485d000f57183fe8..7c14294421113907e360688a340c455c71f74004 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