main.py

import os

# import subprocess
# import pygame
# pygame.init()

###############################################################################
## TEST #######################################################################
###############################################################################
from src.crazy_env import CrazyEnv

###############################################################################
## CRAZY TORCH  ###############################################################
###############################################################################
from src.crazy_torch.crazy_stable_dqn import crazy_stable_dqn_training
from src.crazy_torch.crazy_dqn import CrazyDQN

###############################################################################
## CRAZY FLOW  ################################################################
###############################################################################

if __name__ == "__main__":
    # Change working directory
    abspath = os.path.abspath(__file__)
    dname = os.path.dirname(abspath)
    os.chdir(dname)

    # crazy_cmd = [f"python", "crazy_driver.py"]
    # subprocess.run(crazy_cmd, cwd="crazy_driver_1")

    # CRAZY TORCH
    CrazyDQN = CrazyDQN()
    CrazyDQN.train()

    # CRAZY FLOW