__init__.py

# coding=utf-8

import sys
import os

from ActorCritic.BatchActorCriticAgent import BatchActorCriticAgent
from ActorCritic.OnlineActorCriticAgent import OnlineActorCriticAgent
from ActorCritic.OnlineTwoInputAdvantageActorCriticAgent import OnlineTwoInputAdvantageActorCriticAgent
from ActorCritic.ActorCriticBrainSplitNetwork import (build_actor_policy_graph, build_two_input_critic_graph,
                                                      build_critic_graph, actor_train, critic_train)
from ActorCritic.ActorCriticBrainSharedNetwork import build_actor_critic_shared_graph

name = "ActorCritic"

def solve_PYTHONPATH():
    """Solve: broken import path problem occuring when script are invocated from command line

    This problematic behavior is a side effect of importing module/package outside the current package tree.
    This is not problematic when your working in a controled environment (aka a IDE with hardcoded PYTHONPATH).
    However, when your script will be execute from the command line (outside your IDE) all hell will break loose.

    (CRITICAL) todo --> appended to each package __init__ containing module that will be invocated from command line:
    (nice to have) todo:refactor --> make one, root level, parameterizable function:
                                                                see DRLimplementation/BasicPolicyGradient/__init__.py
    """
    absolute_current_dir = os.getcwd()
    print(":: On INIT - cwd: {}".format(absolute_current_dir))

    if os.path.basename(absolute_current_dir) != "DRLimplementation":
        # meaning the module invocating the script is not at project root level

        while True:
            absolute_parent_dir, curent_dir = os.path.split(absolute_current_dir)

            if os.path.basename(absolute_parent_dir) == "DRLimplementation":
                assert os.path.exists(absolute_parent_dir), "Something is wrong with path resolution"
                sys.path.insert(0, absolute_parent_dir)
                print(":: insert to sys.path: {}".format(absolute_parent_dir))
                break
            else:
                absolute_current_dir = absolute_parent_dir


solve_PYTHONPATH()