Attempt to fix help(MyDistribution)
# Copyright (c) 2017-2019 Uber Technologies, Inc.
# SPDX-License-Identifier: Apache-2.0
+import functools
from abc import ABCMeta, abstractmethod
from pyro.distributions.score_parts import ScoreParts
@@ -16,6 +17,10 @@ class DistributionMeta(ABCMeta):
return result
return super().__call__(*args, **kwargs)
+ @property
+ def __wrapped__(cls):
+ return functools.partial(cls.__init__, None)
+
class Distribution(metaclass=DistributionMeta):
"""
