Update black version to 22.3>= to fix issue with newer version of Click (https://github.com/pallets/click/issues/2225).
Also modify test_loss.py to change np.testing.assert_equal(ce_loss.numpy(), ces_loss.numpy()) to np.testing.assert_almost_equal(ce_loss.numpy(), ces_loss.numpy()) to deal with newer torch code.
Related issue(s)
Fixes # (issue)
Test plan
Checklist
Need help on these? Just ask!
[ ] I have read the CONTRIBUTING document.
[ ] I have updated the documentation accordingly.
[ ] I have added tests to cover my changes.
[ ] I have run tox -e complex and/or tox -e spark if appropriate.
Description of proposed changes
Update black version to 22.3>= to fix issue with newer version of Click (https://github.com/pallets/click/issues/2225). Also modify
test_loss.pyto changenp.testing.assert_equal(ce_loss.numpy(), ces_loss.numpy())tonp.testing.assert_almost_equal(ce_loss.numpy(), ces_loss.numpy())to deal with newer torch code.Related issue(s)
Fixes # (issue)
Test plan
Checklist
Need help on these? Just ask!
tox -e complexand/ortox -e sparkif appropriate.