Removing the type ignores

Author: emersodb

fl4health/utils/functions.py

@@ -19,7 +19,7 @@ class BernoulliSample(torch.autograd.Function):
     """
 
     @staticmethod
-    def forward(bernoulli_probs: torch.Tensor) -> torch.Tensor:  # type: ignore
+    def forward(bernoulli_probs: torch.Tensor) -> torch.Tensor:
         return torch.bernoulli(input=bernoulli_probs)
 
     @staticmethod