env_consts.py

import os

TEST_INPUT_DIR = None
TEST_OUTPUT_DIR = None
THIS_FILE_DIR = os.path.dirname(os.path.abspath(__file__))
CKPT_PATH = os.path.join(THIS_FILE_DIR, "resources", "only_weights_87-172000.ckpt")
RUN_CONFIG_PATH = os.path.join(THIS_FILE_DIR, "resources", "run_config.json")

OUTPUT_PROT_PATH = os.path.join(THIS_FILE_DIR, "predicted_protein_out.pdb")
OUTPUT_LIG_PATH = os.path.join(THIS_FILE_DIR, "predicted_lig_out.sdf")