# Buggy code
def fit(self, X, y=None):
    """Store all observed homology dimensions in
    :attr:`homology_dimensions_` and, for each dimension separately,
    store evenly sample filtration parameter values in :attr:`samplings_`.
    Then, return the estimator.

    This method is here to implement the usual scikit-learn API and hence
    work in pipelines.

    Parameters
    ----------
    X : ndarray of shape (n_samples, n_features, 3)
        Input data. Array of persistence diagrams, each a collection of
        triples [b, d, q] representing persistent topological features
        through their birth (b), death (d) and homology dimension (q).
        It is important that, for each possible homology dimension, the
        number of triples for which q equals that homology dimension is
        constants across the entries of X.

    y : None
        There is no need for a target in a transformer, yet the pipeline
        API requires this parameter.

    Returns
    -------
    self : object

    """
    X = check_diagrams(X)
    validate_params(self.get_params(), self._hyperparameters, exclude=["n_jobs"])

    self.homology_dimensions_ = sorted(list(set(X[0, :, 2])))
    self._n_dimensions = len(self.homology_dimensions_)
    self._samplings, self._step_size = _bin(X, metric="heat", n_bins=self.n_bins)
    self.samplings_ = {dim: s.flatten() for dim, s in self._samplings.items()}

    return self


# Traceback (TypeError)
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-5-5e233492b878> in <module>
----> 1 Amplitude(metric="persistence_image").fit_transform(Xd)

~\Workspace\giotto-tda_ulupo\gtda\utils\_docs.py in fit_transform_wrapper(*args, **kwargs)
104         @wraps(original_fit_transform)
105         def fit_transform_wrapper(*args, **kwargs):
--> 106             return original_fit_transform(*args, **kwargs)
107         fit_transform_wrapper.__doc__ = \
108             make_fit_transform_docs(fit_docs, transform_docs)

~\Anaconda3\envs\gtda-ulupo-py38\lib\site-packages\sklearn\base.py in fit_transform(self, X, y, **fit_params)
688         if y is None:
689             # fit method of arity 1 (unsupervised transformation)
--> 690             return self.fit(X, **fit_params).transform(X)
691         else:
692             # fit method of arity 2 (supervised transformation)

~\Workspace\giotto-tda_ulupo\gtda\diagrams\features.py in fit(self, X, y)
273         if self.metric == 'persistence_image':
274             self.effective_metric_params_['weights'] = \
--> 275                 _calculate_weights(X, **self.effective_metric_params_)
276
277         return self

TypeError: _calculate_weights() missing 1 required positional argument: 'weight_function'

# Fixed code
def fit(self, X, y=None):
    """Store all observed homology dimensions in
    :attr:`homology_dimensions_` and, for each dimension separately,
    store evenly sample filtration parameter values in :attr:`samplings_`.
    Then, return the estimator.

    This method is here to implement the usual scikit-learn API and hence
    work in pipelines.

    Parameters
    ----------
    X : ndarray of shape (n_samples, n_features, 3)
        Input data. Array of persistence diagrams, each a collection of
        triples [b, d, q] representing persistent topological features
        through their birth (b), death (d) and homology dimension (q).
        It is important that, for each possible homology dimension, the
        number of triples for which q equals that homology dimension is
        constants across the entries of X.

    y : None
        There is no need for a target in a transformer, yet the pipeline
        API requires this parameter.

    Returns
    -------
    self : object

    """
    X = check_diagrams(X)
    validate_params(self.get_params(), self._hyperparameters, exclude=["n_jobs"])

    # Find the unique homology dimensions in the 3D array X passed to `fit`
    # assuming that they can all be found in its zero-th entry
    homology_dimensions_fit = np.unique(X[0, :, 2])
    self.homology_dimensions_ = _homology_dimensions_to_sorted_ints(
        homology_dimensions_fit
    )
    self._n_dimensions = len(self.homology_dimensions_)

    self._samplings, self._step_size = _bin(
        X, "heat", n_bins=self.n_bins, homology_dimensions=self.homology_dimensions_
    )
    self.samplings_ = {dim: s.flatten() for dim, s in self._samplings.items()}

    return self