From 30af410ac6673b4a6b7f9ade551531c3b4b3c577 Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Wed, 25 Apr 2018 15:37:57 +0200
Subject: [PATCH] Renamed assignment_collection submodule to simp (for
 simplifications)

---
 __init__.py                                            |  5 +++--
 fd/finitedifferences.py                                |  5 ++---
 kernelcreation.py                                      | 10 +++++-----
 {assignment_collection => simp}/__init__.py            |  6 +++---
 .../assignment_collection.py                           |  0
 {assignment_collection => simp}/simplifications.py     |  2 +-
 .../simplificationstrategy.py                          |  2 +-
 7 files changed, 15 insertions(+), 15 deletions(-)
 rename {assignment_collection => simp}/__init__.py (65%)
 rename {assignment_collection => simp}/assignment_collection.py (100%)
 rename {assignment_collection => simp}/simplifications.py (97%)
 rename {assignment_collection => simp}/simplificationstrategy.py (98%)

diff --git a/__init__.py b/__init__.py
index 4ba13340e..0c574e912 100644
--- a/__init__.py
+++ b/__init__.py
@@ -5,11 +5,11 @@ from .data_types import TypedSymbol
 from .slicing import make_slice
 from .kernelcreation import create_kernel, create_indexed_kernel
 from .display_utils import show_code, to_dot
-from .assignment_collection import AssignmentCollection
+from .simp import AssignmentCollection
 from .assignment import Assignment
 from .sympyextensions import SymbolCreator
 from .datahandling import create_data_handling
-from . kernel_decorator import kernel
+from .kernel_decorator import kernel
 
 __all__ = ['Field', 'FieldType', 'fields',
            'TypedSymbol',
@@ -21,3 +21,4 @@ __all__ = ['Field', 'FieldType', 'fields',
            'SymbolCreator',
            'create_data_handling',
            'kernel']
+
diff --git a/fd/finitedifferences.py b/fd/finitedifferences.py
index f39d4c0c7..4b584779b 100644
--- a/fd/finitedifferences.py
+++ b/fd/finitedifferences.py
@@ -2,10 +2,9 @@ import numpy as np
 import sympy as sp
 from typing import Union, Optional
 
-from pystencils.assignment_collection import AssignmentCollection
-from pystencils.field import Field
+from pystencils import Field, AssignmentCollection
+from pystencils.fd import Diff
 from pystencils.sympyextensions import fast_subs
-from pystencils.fd.derivative import Diff
 
 
 FieldOrFieldAccess = Union[Field, Field.Access]
diff --git a/kernelcreation.py b/kernelcreation.py
index 45eebc34c..17d804b2d 100644
--- a/kernelcreation.py
+++ b/kernelcreation.py
@@ -1,10 +1,10 @@
 from types import MappingProxyType
 import sympy as sp
-from pystencils.assignment import Assignment
-from pystencils.astnodes import LoopOverCoordinate, Conditional, Block, SympyAssignment
-from pystencils.assignment_collection import AssignmentCollection
-from pystencils.gpucuda.indexing import indexing_creator_from_params
-from pystencils.transformations import remove_conditionals_in_staggered_kernel
+from .assignment import Assignment
+from .astnodes import LoopOverCoordinate, Conditional, Block, SympyAssignment
+from .simp.assignment_collection import AssignmentCollection
+from .gpucuda.indexing import indexing_creator_from_params
+from .transformations import remove_conditionals_in_staggered_kernel
 
 
 def create_kernel(assignments, target='cpu', data_type="double", iteration_slice=None, ghost_layers=None,
diff --git a/assignment_collection/__init__.py b/simp/__init__.py
similarity index 65%
rename from assignment_collection/__init__.py
rename to simp/__init__.py
index 7648bbab8..0e8a85d1f 100644
--- a/assignment_collection/__init__.py
+++ b/simp/__init__.py
@@ -1,6 +1,6 @@
-from pystencils.assignment_collection.assignment_collection import AssignmentCollection
-from pystencils.assignment_collection.simplificationstrategy import SimplificationStrategy
-from pystencils.assignment_collection.simplifications import sympy_cse, sympy_cse_on_assignment_list, \
+from .assignment_collection import AssignmentCollection
+from .simplificationstrategy import SimplificationStrategy
+from .simplifications import sympy_cse, sympy_cse_on_assignment_list, \
     apply_to_all_assignments, apply_on_all_subexpressions, subexpression_substitution_in_existing_subexpressions, \
     subexpression_substitution_in_main_assignments, add_subexpressions_for_divisions
 
diff --git a/assignment_collection/assignment_collection.py b/simp/assignment_collection.py
similarity index 100%
rename from assignment_collection/assignment_collection.py
rename to simp/assignment_collection.py
diff --git a/assignment_collection/simplifications.py b/simp/simplifications.py
similarity index 97%
rename from assignment_collection/simplifications.py
rename to simp/simplifications.py
index 6d6de1414..2cb4dd303 100644
--- a/assignment_collection/simplifications.py
+++ b/simp/simplifications.py
@@ -1,7 +1,7 @@
 import sympy as sp
 from typing import Callable, List
 from pystencils.assignment import Assignment
-from pystencils.assignment_collection.assignment_collection import AssignmentCollection
+from pystencils.simp.assignment_collection import AssignmentCollection
 from pystencils.sympyextensions import subs_additive
 
 AC = AssignmentCollection
diff --git a/assignment_collection/simplificationstrategy.py b/simp/simplificationstrategy.py
similarity index 98%
rename from assignment_collection/simplificationstrategy.py
rename to simp/simplificationstrategy.py
index d87b1ee1b..f80734bab 100644
--- a/assignment_collection/simplificationstrategy.py
+++ b/simp/simplificationstrategy.py
@@ -1,7 +1,7 @@
 import sympy as sp
 from collections import namedtuple
 from typing import Callable, Any, Optional, Sequence
-from pystencils.assignment_collection.assignment_collection import AssignmentCollection
+from pystencils.simp.assignment_collection import AssignmentCollection
 
 
 class SimplificationStrategy:
-- 
GitLab