TypeError: Invalid keyword argument(s) in `compile`: {'jit_compile'}

[aik7]

When I tried to use my exarl conda environment in Perlmutter, I get the following error when I run:

Note: I am using the fixBroken branch.

python exarl/driver/ --env Pendulum-v0 --n_episodes 50 --n_steps 200 --learner_procs 1 --workflow sync --agent DDPG-v0 --model_type AC

Traceback (most recent call last):
  File "/global/homes/a/aik7/anaconda3/envs/exarl/lib/python3.7/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/global/homes/a/aik7/anaconda3/envs/exarl/lib/python3.7/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "exarl/driver/__main__.py", line 30, in <module>
    exa_learner = exarl.ExaLearner()
  File "/global/u2/a/aik7/EXARL/exarl/base/learner_base.py", line 46, in __init__
    self.agent, self.env, self.workflow = self.make(agent_id, env_id, workflow_id)
  File "/global/u2/a/aik7/EXARL/exarl/base/learner_base.py", line 95, in make
    agent = exarl.agents.make(agent_id, env=env, is_learner=ExaComm.is_learner())
  File "/global/u2/a/aik7/EXARL/exarl/agents/registration.py", line 98, in make
    return registry.make(id, **kwargs)
  File "/global/u2/a/aik7/EXARL/exarl/agents/registration.py", line 58, in make
    agent = spec.make(**kwargs)
  File "/global/u2/a/aik7/EXARL/exarl/agents/registration.py", line 43, in make
    agent = cls(**_kwargs)
  File "/global/u2/a/aik7/EXARL/exarl/agents/agent_vault/ddpg.py", line 75, in __init__
    self.actor.init_model()
  File "/global/u2/a/aik7/EXARL/exarl/agents/models/tf_model.py", line 194, in init_model
    self._compile()
  File "/global/u2/a/aik7/EXARL/exarl/agents/models/tf_model.py", line 146, in _compile
    self._model.compile(loss=self.loss, optimizer=self.optimizer, jit_compile=self.enable_xla)
  File "/global/homes/a/aik7/anaconda3/envs/exarl/lib/python3.7/site-packages/tensorflow/python/keras/engine/training.py", line 534, in compile
    self._validate_compile(optimizer, metrics, **kwargs)
  File "/global/homes/a/aik7/anaconda3/envs/exarl/lib/python3.7/site-packages/tensorflow/python/keras/engine/training.py", line 2520, in _validate_compile
    (invalid_kwargs,))
TypeError: Invalid keyword argument(s) in `compile`: {'jit_compile'}