Skip to content

Commit 6db9507

Browse files
authored
Avoid pip wheel caching (#4)
1 parent bf080c0 commit 6db9507

File tree

1 file changed

+21
-3
lines changed

1 file changed

+21
-3
lines changed

setup.py

Lines changed: 21 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,10 +2,27 @@
22
import sys
33
from datetime import datetime, MAXYEAR
44
from collections import namedtuple
5+
import setuptools
56

6-
from setuptools import setup
7-
from setuptools.command.install import install
7+
try:
8+
from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
89

10+
class bdist_wheel(_bdist_wheel):
11+
def run(self):
12+
message = "\n".join(
13+
[
14+
"This is an expected error. Building wheel is disabled "
15+
"for the deprecated sklearn PyPI package to avoid pip caching.",
16+
"For more details about the sklearn PyPI package deprecation, see:",
17+
"https://github.com/scikit-learn/sklearn-pypi-package",
18+
]
19+
)
20+
raise setuptools.errors.ClassError(message)
21+
22+
cmdclass = {"bdist_wheel": bdist_wheel}
23+
24+
except ImportError:
25+
cmdclass = {}
926

1027
with open("README.md") as f:
1128
LONG_DESCRIPTION = f.read()
@@ -101,10 +118,11 @@ def maybe_raise_error(checked_datetime):
101118

102119
maybe_raise_error(checked_datetime)
103120

104-
setup(
121+
setuptools.setup(
105122
description="deprecated sklearn package, use scikit-learn instead",
106123
long_description=LONG_DESCRIPTION,
107124
long_description_content_type="text/markdown",
108125
name="sklearn",
109126
version="0.0.post1",
127+
cmdclass=cmdclass,
110128
)

0 commit comments

Comments
 (0)