From 731e40a4dd9d1ba1cde0796f8bd39aa678d2bd0f Mon Sep 17 00:00:00 2001 From: Max Halford Date: Wed, 2 Aug 2023 10:19:44 +0200 Subject: [PATCH] ruff --- river/proba/__init__.py | 3 +-- river/proba/gaussian.py | 6 ++---- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/river/proba/__init__.py b/river/proba/__init__.py index 49536c626e..c3e399534a 100644 --- a/river/proba/__init__.py +++ b/river/proba/__init__.py @@ -3,8 +3,7 @@ from . import base from .beta import Beta -from .gaussian import Gaussian -from .gaussian import MultivariateGaussian +from .gaussian import Gaussian, MultivariateGaussian from .multinomial import Multinomial __all__ = ["base", "Beta", "Gaussian", "Multinomial", "MultivariateGaussian"] diff --git a/river/proba/gaussian.py b/river/proba/gaussian.py index 86e37a596c..2c8f51ae28 100644 --- a/river/proba/gaussian.py +++ b/river/proba/gaussian.py @@ -1,15 +1,13 @@ from __future__ import annotations import math -import pytest -import warnings import numpy as np import pandas as pd +import pytest from scipy.stats import multivariate_normal -from river import covariance -from river import stats +from river import covariance, stats from river.proba import base __all__ = ["Gaussian", "MultivariateGaussian"]