diff --git a/modl/dict_fact.py b/modl/dict_fact.py index 9f1dbc4..738e412 100644 --- a/modl/dict_fact.py +++ b/modl/dict_fact.py @@ -181,9 +181,9 @@ def _init(self, X): order='F') self.B_ = np.zeros((self.n_components, n_cols), order="F") - self.counter_ = np.zeros(n_cols + 1, dtype='int') + self.counter_ = np.zeros(n_cols + 1, dtype=int) - self.n_iter_ = np.zeros(1, dtype='long') + self.n_iter_ = np.zeros(1, dtype=long) self.code_ = np.zeros((self.n_samples_, self.n_components)) @@ -587,4 +587,4 @@ def _update_dict_slow(self, subset, D_range): def _callback(self): if self.callback is not None: - self.callback(self) \ No newline at end of file + self.callback(self) diff --git a/setup.py b/setup.py index 7c55dea..257e4ca 100755 --- a/setup.py +++ b/setup.py @@ -9,7 +9,7 @@ DISTNAME = 'modl' DESCRIPTION = "Masked Online Dictionary Learning in Python" -LONG_DESCRIPTION = open('README.rst').read() +LONG_DESCRIPTION = open('README.md').read() MAINTAINER = 'Arthur Mensch' MAINTAINER_EMAIL = 'arthur.mensch@m4x.org' URL = 'https://github.com/arthurmensch/modl'