if __name__ == "__main__":
// set up logger
logging.getLogger().setLevel(logging.INFO)
// parse args
parser = argparse.ArgumentParser(description="Run a grasping policy on an example image")
parser.add_argument("--depth_image", type=str, default=None, help="path to a test depth image stored as a .npy file")
After Change
if __name__ == "__main__":
// set up logger
logger = get_logger("ExamplePolicy", log_stream=sys.stdout)
// parse args
parser = argparse.ArgumentParser(description="Run a grasping policy on an example image")
parser.add_argument("--depth_image", type=str, default=None, help="path to a test depth image stored as a .npy file")