Apply suggestions from code review

Lightning-AI · Sep 7, 2021 · 8314b3b · 8314b3b
1 parent 03b66da
commit 8314b3b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/requirements/test.txt b/requirements/test.txt
@@ -10,7 +10,6 @@ twine>=3.2
 isort>=5.0
 mypy>=0.790
 phmdoctest>=1.1.1
-PyYAML>=5.1
 
 cloudpickle>=1.3
 scikit-learn>=0.24

diff --git a/tests/helpers/testers.py b/tests/helpers/testers.py
@@ -155,9 +155,6 @@ def _class_test(
  if check_scriptable:
  torch.jit.script(metric)
 
- # check that metrics can be parsed with yaml
- yaml.dump(metric)
-
  # move to device
  metric = metric.to(device)
  preds = preds.to(device)