Merge pull request #18 from HumanCompatibleAI/type-conversion-fix

Explicit type conversion for str->int in reddit engagement
HumanCompatibleAI · Apr 17, 2024 · ae88eb0 · ae88eb0
2 parents f15a4aa + 5a5e77e
commit ae88eb0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/integration_test/integration_test.py b/examples/integration_test/integration_test.py
@@ -96,8 +96,8 @@ def test_rank_reddit(url):
                 r"https?://(?:[-\w.]|(?:%[\da-fA-F]{2}))+", row["text"]
             )
             engagements = RedditEngagements(
-                upvote=row["upvotes"],
-                downvote=row["downvotes"],
+                upvote=int(float(row["upvotes"])),
+                downvote=int(float(row["downvotes"])),
                 comment=0,
                 award=0,
             )