From bf8b0d132b892460e21437b35964cecdabbce443 Mon Sep 17 00:00:00 2001 From: Haoyin Xu <haoyinxu@gmail.com> Date: Mon, 6 Nov 2023 13:04:54 -0500 Subject: [PATCH] FIX remove multiview dataset import & fix entropy --- sktree/datasets/__init__.py | 1 - sktree/tests/test_honest_forest.py | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/sktree/datasets/__init__.py b/sktree/datasets/__init__.py index 74c770d6a..d49e247a9 100644 --- a/sktree/datasets/__init__.py +++ b/sktree/datasets/__init__.py @@ -1,2 +1 @@ from .hyppo import make_quadratic_classification -from .multiview import make_gaussian_mixture, make_joint_factor_model diff --git a/sktree/tests/test_honest_forest.py b/sktree/tests/test_honest_forest.py index 71cd9a42a..a591872e2 100644 --- a/sktree/tests/test_honest_forest.py +++ b/sktree/tests/test_honest_forest.py @@ -1,5 +1,6 @@ import numpy as np import pytest +from scipy.stats import entropy from numpy.testing import assert_allclose, assert_array_almost_equal from sklearn import datasets from sklearn.metrics import accuracy_score, r2_score, roc_auc_score