code: def main(args):
      cwd = get_original_cwd()
      os.chdir(cwd)
      print("Start training Entity Model")
      entity_train(args)
      print("Start training Relation Model")
      relation_train(args)
  
file_path: DeepKE/example/triple/PURE/run.py
imports: import os
  import sys
  import argparse
  import hydra
  from hydra.utils import get_original_cwd
  import torch
  import logging
  import tqdm
  import wandb
  from models.const import *
  from models.data_structures import *
  from models.entityModels import *
  from models.entityUtils import *
  from models.relationModels import *
  from models.relationUtils import *
  from run_entity import *
  from run_relation import *
label_desc: Use image node_id 3 for calling a global function with example usage: main(args) without return types