Changed AdamW test condition (1add21f9) · Commits · 钟慕尧 / deepchem

deepchem/models/tests/test_optimizers.py

+2 −2

Original line number	Diff line number	Diff line
		@@ -39,7 +39,7 @@ class TestOptimizers(unittest.TestCase):
		torchopt = opt._create_pytorch_optimizer(params)
		assert isinstance(torchopt, torch.optim.Adam)

		@unittest.skipIf(not has_tensorflow, 'TensorFlow is not installed')
		@unittest.skipIf(not has_tensorflow_addons, 'TensorFlow Addons is not installed')
		def test_adamw_tf(self):
		"""Test creating an AdamW optimizer."""
		opt = optimizers.AdamW(learning_rate=0.01)
		@@ -47,7 +47,7 @@ class TestOptimizers(unittest.TestCase):
		tfopt = opt._create_tf_optimizer(global_step)
		assert isinstance(tfopt, tfa.optimizers.AdamW)

		@unittest.skipIf(not has_pytorch, 'PyTorch is not installed')
		@unittest.skipIf(not has_tensorflow_addons, 'TensorFlow Addons is not installed')
		def test_adamw_pytorch(self):
		"""Test creating an AdamW optimizer."""
		opt = optimizers.AdamW(learning_rate=0.01)