GitLab

```
./openfold/utils/loss.py:1498:21: F821 undefined name 'logging'
                    logging.warning(f"{loss_name} loss is NaN. Skipping...")
                    ^
./openfold/utils/feats.py:180:15: F821 undefined name 'math'
        2.0 / math.pi
              ^
./openfold/utils/feats.py:190:31: F821 undefined name 'tf'
        deletion_mean_value = tf.atan(batch["cluster_deletion_mean"] / 3.0) * (
                              ^
./openfold/utils/feats.py:196:17: F821 undefined name 'tf'
                tf.expand_dims(deletion_mean_value, axis=-1),
                ^
./openfold/utils/feats.py:200:35: F821 undefined name 'protein'
    if "extra_deletion_matrix" in protein:
                                  ^
./openfold/utils/feats.py:201:39: F821 undefined name 'tf'
        batch["extra_has_deletion"] = tf.clip_by_value(
                                      ^
./openfold/utils/feats.py:204:41: F821 undefined name 'tf'
        batch["extra_deletion_value"] = tf.atan(
                                        ^
```