Fixing sklearn errors 1

[codecov[bot]]

Codecov Report

Base: 73.29% // Head: 76.08% // Increases project coverage by +2.78% :tada:

Coverage data is based on head (9617793) compared to base (4ef252a). Patch coverage: 90.90% of modified lines in pull request are covered.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #620 +/- ## ========================================== + Coverage 73.29% 76.08% +2.78% ========================================== Files 67 67 Lines 4794 4804 +10 ========================================== + Hits 3514 3655 +141 + Misses 1280 1149 -131 ``` | Flag | Coverage Δ | | |---|---|---| | python | `76.08% <90.90%> (+2.78%)` | :arrow_up: | Flags with carried forward coverage won't be shown. [Click here](https://docs.codecov.io/docs/carryforward-flags?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout#carryforward-flags-in-the-pull-request-comment) to find out more. | [Impacted Files](https://codecov.io/gh/AIStream-Peelout/flow-forecast/pull/620?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout) | Coverage Δ | | |---|---|---| | [flood\_forecast/trainer.py](https://codecov.io/gh/AIStream-Peelout/flow-forecast/pull/620/diff?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout#diff-Zmxvb2RfZm9yZWNhc3QvdHJhaW5lci5weQ==) | `88.88% <87.50%> (+20.75%)` | :arrow_up: | | [tests/test\_multitask\_decoder.py](https://codecov.io/gh/AIStream-Peelout/flow-forecast/pull/620/diff?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout#diff-dGVzdHMvdGVzdF9tdWx0aXRhc2tfZGVjb2Rlci5weQ==) | `88.88% <100.00%> (+0.65%)` | :arrow_up: | | [flood\_forecast/preprocessing/pytorch\_loaders.py](https://codecov.io/gh/AIStream-Peelout/flow-forecast/pull/620/diff?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout#diff-Zmxvb2RfZm9yZWNhc3QvcHJlcHJvY2Vzc2luZy9weXRvcmNoX2xvYWRlcnMucHk=) | `81.95% <0.00%> (+0.37%)` | :arrow_up: | | [flood\_forecast/custom/custom\_opt.py](https://codecov.io/gh/AIStream-Peelout/flow-forecast/pull/620/diff?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout#diff-Zmxvb2RfZm9yZWNhc3QvY3VzdG9tL2N1c3RvbV9vcHQucHk=) | `63.93% <0.00%> (+0.54%)` | :arrow_up: | | [flood\_forecast/transformer\_xl/dsanet.py](https://codecov.io/gh/AIStream-Peelout/flow-forecast/pull/620/diff?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout#diff-Zmxvb2RfZm9yZWNhc3QvdHJhbnNmb3JtZXJfeGwvZHNhbmV0LnB5) | `93.29% <0.00%> (+57.21%)` | :arrow_up: | Help us with your feedback. Take ten seconds to tell us [how you rate us](https://about.codecov.io/nps?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout). Have a feature suggestion? [Share it here.](https://app.codecov.io/gh/feedback/?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=AIStream-Peelout)

:umbrella: View full report at Codecov.
:loudspeaker: Do you have feedback about the report comment? Let us know in this issue.