Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def reset(m):
if isinstance(m, LinearBayesianLayer):
m.reset_parameters()
self.model.apply(reset)