From 1f85bf2cc0ccf1911efb327be02985ad16ae8d21 Mon Sep 17 00:00:00 2001
From: Matteo Bettini <mb2389@cl.cam.ac.uk>
Date: Wed, 8 Nov 2023 09:48:03 +0000
Subject: [PATCH] remove assert

---
 benchmarl/experiment/experiment.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/benchmarl/experiment/experiment.py b/benchmarl/experiment/experiment.py
index 09986b5b..ba43a3df 100644
--- a/benchmarl/experiment/experiment.py
+++ b/benchmarl/experiment/experiment.py
@@ -8,7 +8,6 @@
 
 import importlib
 
-import math
 import os
 import time
 from collections import OrderedDict
@@ -641,7 +640,6 @@ def _optimizer_loop(self, group: str) -> TensorDictBase:
 
                 grad_norm = self._grad_clip(optimizer)
 
-                assert not math.isnan(grad_norm)
                 training_td.set(
                     f"grad_norm_{loss_name}",
                     torch.tensor(grad_norm, device=self.config.train_device),