Change comparison in partitions to include equals

[JAEarly]

Description of changes:

A warning is raised during partitioning when the number of canonical nodes is equal to the number of samples. It is my understanding that this should be valid (one sample per node), and that issues only occur when the number of samples is less than the number of canonical nodes. Apologies if my understanding is incorrect here.

The simple fix is to the change the comparison to be >= rather than >

Merge Checklist:

Put an x without space in the boxes that apply. If you are unsure about any checklist, please don't hesitate to ask. We are here to help! This is simply a reminder of what we are going to look for before merging your pull request.

General

• I have read the contributor guidelines
• This is a documentation change or typo fix. If so, skip the rest of this checklist.
• I certify that the changes I am introducing will be backward compatible, and I have discussed concerns about this, if any, with the MosaicML team.
• I have updated any necessary documentation, including README and API docs (if appropriate).

Tests

• I ran pre-commit on my change. (check out the pre-commit section of prerequisites)
• I have added tests that prove my fix is effective or that my feature works (if appropriate).
• I ran the tests locally to make sure it pass. (check out testing)
• I have added unit and/or integration tests as appropriate to ensure backward compatibility of the changes.

[JAEarly]

I didn't realise this would break the tests (I made the change in the browser). If this change makes sense I'll happily go and alter the tests as appropriate.

[JAEarly]

I've now updated the code such that the tests pass but the warning is not raised for the edge case: num_canonical_nodes == num_samples and num_canonical_nodes > num_physical_nodes.

[snarayan21]

Hey! Apologies for getting to this late, but just a small change and should be good to go. Great catch!

[snarayan21]

lgtm!