OrdinalCategoricalBucketer test fails

[orchardbirds]

================================== FAILURES ==================================
____________________________ test_encoding_method ____________________________

df =       EDUCATION  MARRIAGE  LIMIT_BAL  BILL_AMT1  default pet_ownership
0             1         2   400000.0   201800.0...        0       no pets
5999          2         1   410000.0    71532.0        1     dog lover

[6000 rows x 6 columns]

    def test_encoding_method(df):
        """Test the encoding method."""
        X = df[["EDUCATION", "default"]]
        y = df["default"]

        ocb = OrdinalCategoricalBucketer(tol=0.03, variables=["EDUCATION"], encoding_method="frequency")
        ocb.fit(X, y)

        assert ocb.features_bucket_mapping_.get("EDUCATION").map == {1: 1, 2: 0, 3: 2}

        ocb = OrdinalCategoricalBucketer(tol=0.03, variables=["EDUCATION"], encoding_method="ordered")
        ocb.fit(X, y)

>       assert ocb.features_bucket_mapping_.get("EDUCATION").map == {1: 2, 2: 0, 3: 1}
E       assert {1: 0, 2: 2, 3: 1} == {1: 2, 2: 0, 3: 1}
E         Omitting 1 identical items, use -vv to show
E         Differing items:
E         {1: 0} != {1: 2}
E         {2: 2} != {2: 0}
E         Use -v to get the full diff

tests/test_bucketer_OrdinalCategoricalBucketer.py:66: AssertionError