Fix lint error

Signed-off-by: Ivan Krizanic <ikrizanic75@gmail.com>
feast-dev · Jul 20, 2022 · b73d640 · b73d640
1 parent 245bd03
commit b73d640
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sdk/python/tests/unit/infra/test_key_encoding_utils.py b/sdk/python/tests/unit/infra/test_key_encoding_utils.py
@@ -9,14 +9,14 @@ def test_serialize_entity_key():
     # Should be fine
     serialize_entity_key(
         EntityKeyProto(
-            join_keys=["user"], entity_values=[ValueProto(int64_val=int(2 ** 15))]
+            join_keys=["user"], entity_values=[ValueProto(int64_val=int(2**15))]
         ),
         entity_key_serialization_version=2,
     )
     # True int64, but should also be fine.
     serialize_entity_key(
         EntityKeyProto(
-            join_keys=["user"], entity_values=[ValueProto(int64_val=int(2 ** 31))]
+            join_keys=["user"], entity_values=[ValueProto(int64_val=int(2**31))]
         ),
         entity_key_serialization_version=2,
     )
@@ -25,6 +25,6 @@ def test_serialize_entity_key():
     with pytest.raises(BaseException):
         serialize_entity_key(
             EntityKeyProto(
-                join_keys=["user"], entity_values=[ValueProto(int64_val=int(2 ** 31))]
+                join_keys=["user"], entity_values=[ValueProto(int64_val=int(2**31))]
             ),
         )