diff --git a/docs/freqai.md b/docs/freqai.md index de321b787..adf64dd07 100644 --- a/docs/freqai.md +++ b/docs/freqai.md @@ -56,6 +56,9 @@ Use `pip` to install the prerequisites with: pip install -r requirements-freqai.txt ``` +!!! Note + Catboost will not be installed on arm devices (raspberry, Mac M1, ARM based VPS, ...), since Catboost does not provide wheels for this platform. + ## Running from the example files An example strategy, an example prediction model, and example config can all be found in diff --git a/requirements-freqai.txt b/requirements-freqai.txt index 060a5219d..e4f4a7c17 100644 --- a/requirements-freqai.txt +++ b/requirements-freqai.txt @@ -4,5 +4,5 @@ # Required for freqai scikit-learn==1.1.1 joblib==1.1.0 -catboost==1.0.4 +catboost==1.0.4; platform_machine != 'aarch64' lightgbm==3.3.2 diff --git a/setup.py b/setup.py index 33e35c5ab..dfadbdefa 100644 --- a/setup.py +++ b/setup.py @@ -15,7 +15,7 @@ hyperopt = [ freqai = [ 'scikit-learn', 'joblib', - 'catboost', + 'catboost; platform_machine != "aarch64"', 'lightgbm', ]