Refactor to src/

Author: pockerman

.gitignore

@@ -1,2 +1,2 @@
-preprocessor/__pycache__/
-exceptions/__pycache__/
+src/preprocessor/__pycache__/
+src/exceptions/__pycache__/

src/algorithms/anonymity_a2c_ray.py

@@ -6,8 +6,8 @@
 import ray.rllib.agents.a3c as a3c
 from ray.tune.logger import pretty_print
 from ray.rllib.env.env_context import EnvContext
-from spaces.environment import TimeStep, StepType
-from spaces.observation_space import ObsSpace
+from src.spaces.environment import TimeStep, StepType
+from src.spaces.observation_space import ObsSpace
 
 
 class DataSetEnv(gym.Env):

src/algorithms/trainer.py

@@ -2,7 +2,7 @@
 Trainer
 """
 
-from utils import INFO
+from src.utils import INFO
 from typing import TypeVar
 
 Env = TypeVar("Env")

src/datasets/dataset_wrapper.py

@@ -4,7 +4,7 @@
 import pandas as pd
 import numpy as np
 
-from preprocessor.cleanup_utils import read_csv, replace, change_column_types
+from src.preprocessor.cleanup_utils import read_csv, replace, change_column_types
 
 DS = TypeVar("DS")
 HierarchyBase = TypeVar('HierarchyBase')

src/datasets/datasets_loaders.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from datasets.dataset_wrapper import PandasDSWrapper
+from src.datasets.dataset_wrapper import PandasDSWrapper
 
 
 class MockSubjectsLoader(PandasDSWrapper):
@@ -8,7 +8,7 @@ class MockSubjectsLoader(PandasDSWrapper):
                        "salary": int, "diagnosis": int, "preventative_treatment": str,
                        "mutation_status": int, }
 
-    FILENAME = Path("../data/mocksubjects.csv")
+    FILENAME = Path("../../data/mocksubjects.csv")
 
     FEATURES_DROP_NAMES = ["NHSno", "given_name", "surname", "dob"]
 

src/spaces/action_space.py

@@ -1,6 +1,5 @@
-from typing import List
 from gym.spaces.discrete import Discrete
-from spaces.actions import ActionBase
+from src.spaces.actions import ActionBase
 
 
 class ActionSpace(Discrete):

src/spaces/actions.py

@@ -2,7 +2,7 @@
 import enum
 from typing import List
 
-from utils.hierarchy_base import HierarchyBase
+from src.utils.hierarchy_base import HierarchyBase
 
 
 class ActionType(enum.IntEnum):

src/spaces/environment.py

@@ -3,18 +3,17 @@
 https://github.com/deepmind/dm_env/blob/master/dm_env/_environment.py
 """
 
-import abc
 import copy
 import enum
 import numpy as np
 import pandas as pd
 import torch
-from typing import Any, NamedTuple, Generic, Optional, TypeVar
+from typing import NamedTuple, Generic, Optional, TypeVar
 import multiprocessing as mp
 
-from exceptions.exceptions import Error
-from spaces.actions import ActionBase
-from utils.string_distance_calculator import DistanceType, TextDistanceCalculator
+from src.exceptions.exceptions import Error
+from src.spaces.actions import ActionBase
+from src.utils.string_distance_calculator import DistanceType, TextDistanceCalculator
 
 DataSet = TypeVar("DataSet")
 

src/tests/test_actions.py

@@ -1,7 +1,7 @@
 import unittest
 
-from utils.default_hierarchy import DefaultHierarchy
-from spaces.actions import ActionGeneralize, ActionSuppress
+from src.utils.default_hierarchy import DefaultHierarchy
+from src.spaces import ActionSuppress
 
 
 class TestActions(unittest.TestCase):

src/tests/test_default_hierarchy.py

@@ -1,5 +1,5 @@
 import unittest
-from utils.default_hierarchy import DefaultHierarchy
+from src.utils.default_hierarchy import DefaultHierarchy
 
 
 class TestDefaultHierarchy(unittest.TestCase):