class_docstr: Tests for policy loader.
class_docstr_tok: ["Tests","for","policy","loader","."]
code: def _createModelsOnDisk(self):
      saved_model_dir = os.path.join(self.root_dir, "policy")
      ckpt_dir = os.path.join(self.root_dir, "checkpoint")
      self.train_step.assign(0)
      self.net.var.assign(0)
      saved_at_0_path = os.path.join(saved_model_dir, "000")
      self.saver.save(saved_at_0_path)
      self.train_step.assign(1)
      self.net.var.assign(10)
      ckpt_at_1_path = os.path.join(ckpt_dir, "001")
      self.saver.save_checkpoint(ckpt_at_1_path)
      return saved_at_0_path, ckpt_at_1_path
  
imports: import os
  import numpy
  import tensorflow.compat.v2
  from tf_agents.networks import network
  from tf_agents.policies import greedy_policy
  from tf_agents.policies import policy_loader
  from tf_agents.policies import policy_saver
  from tf_agents.policies import q_policy
  from tf_agents.specs import tensor_spec
  from tf_agents.trajectories import time_step
  from tf_agents.utils import common
  from tf_agents.utils import test_utils
label_desc: Use image node_id 2 for calling the PolicyLoaderTest obj's underlying member method code with example usage: obj._createModelsOnDisk() and returns: saved_at_0_path,  ckpt_at_1_path