SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[Flax] Update no init test for Flax v0.7.1 (#28735)

Browse Source
This commit is contained in:
Sanchit Gandhi
2024-01-26 18:20:39 +00:00
committed by GitHub
parent abe0289e6d
commit de13a951b3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_flax_common.py
View File

@@ -984,7 +984,7 @@ class FlaxModelTesterMixin:
# Check if we params can be properly initialized when calling init_weights # Check if we params can be properly initialized when calling init_weights
params = model.init_weights(model.key, model.input_shape) params = model.init_weights(model.key, model.input_shape)
self.assertIsInstance(params, FrozenDict) assert isinstance(params, (dict, FrozenDict)), f"params are not an instance of {FrozenDict}"
# Check if all required parmas are initialized # Check if all required parmas are initialized
keys = set(flatten_dict(unfreeze(params)).keys()) keys = set(flatten_dict(unfreeze(params)).keys())
self.assertTrue(all(k in keys for k in model.required_params)) self.assertTrue(all(k in keys for k in model.required_params))