The latest runs of CI tests are failing due to failing imports from sklearn. We should pin the newest version of sklearn to one that works, and eventually update the code to be compatible with the latest version.