code: def test_a2c_reload():
      env = RouletteEnvironment()
      policy = TestPolicy(env)
  
      a2c = dc.rl.A2C(
          env,
          policy,
          max_rollout_length=20,
          optimizer=Adam(learning_rate=0.001),
      )
      a2c.fit(1000)
      action_prob, value = a2c.predict([[0]])
  
      new_a2c = dc.rl.A2C(env, policy, model_dir=a2c._model.model_dir)
      new_a2c.restore()
      action_prob2, value2 = new_a2c.predict([[0]])
  
      assert np.all(action_prob == action_prob2)
      assert value == value2
  
imports: import deepchem
  import numpy
  import pytest
  from deepchem.models.optimizers import Adam
label_desc: Use image node_id 1 for calling a global function with example usage: test_a2c_reload() without return types