Type mismatch for layer.bias

Hi,
I’m working on a CNN implementation in PyTorch, in this PR. I’m getting mypy error in CI Test.
The mypy check fails saying :

deepchem/models/torch_models/cnn.py:208: error: Argument 1 to "constant_" has incompatible type "Optional[Tensor]"; expected "Tensor"
Found 1 error in 1 file (checked 364 source files)
Error: Process completed with exit code 1.

I believe the cause of error is :
In ConvNd layers definition in PyTorch source code here , bias has type Optional[Tensor] .

nn.init.constant_ , expects torch.Tensor argument

Is there any way to fix this mypy error ?