Skip to content

Commit 73b5971

Browse files
fix: Migrate deprecated usages of mock.call_once_with() to mock.assert_called_once_with().
PiperOrigin-RevId: 716168851
1 parent 5b01509 commit 73b5971

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorflow_examples/lite/model_maker/core/data_util/recommendation_dataloader_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def test_download_and_extract_data(self):
3131
with _testutil.patch_download_and_extract_data(self.dataset_dir) as fn:
3232
out_dir = _dl.RecommendationDataLoader.download_and_extract_movielens(
3333
self.download_dir)
34-
fn.called_once_with(self.download_dir)
34+
fn.assert_called_once_with(self.download_dir)
3535
self.assertEqual(out_dir, self.dataset_dir)
3636

3737
def test_generate_movielens_dataset(self):

0 commit comments

Comments
 (0)