Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

clean the warning code

This commit is contained in:
王程浩 2022-03-25 10:00:45 +00:00 committed by cheng-hao-wang
parent 77769f2a41
commit a315a2bece
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/nn/probability/distribution/distribution.py
View File

@ -231,7 +231,7 @@ class Distribution(Cell):
Calculate the broadcast shape of the parameters used during initialization.
"""
broadcast_shape_tensor = None
param_dict = self.parameters['param_dict']
param_dict = self.parameters.get('param_dict')
for value in param_dict.values():
if value is None:
return None