From 4f24d399b87ba7c7a3ea5dc7e2719eab12cea0a9 Mon Sep 17 00:00:00 2001
From: Stefano Pigozzi <me@steffo.eu>
Date: Sat, 4 Feb 2023 05:16:54 +0100
Subject: [PATCH] Convert `DataTuple` to a `collections.namedtuple`

---
 unimore_bda_6/database.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/unimore_bda_6/database.py b/unimore_bda_6/database.py
index 249db99..6304051 100644
--- a/unimore_bda_6/database.py
+++ b/unimore_bda_6/database.py
@@ -5,6 +5,7 @@ import contextlib
 import bson
 import logging
 import itertools
+import collections
 
 from .config import MONGO_HOST, MONGO_PORT, WORKING_SET_SIZE
 
@@ -26,7 +27,7 @@ class Review(t.TypedDict):
 
 Text = str
 Category = str
-DataTuple = tuple[Text, Category]
+DataTuple = collections.namedtuple("DataTuple", ["text", "category"])
 DataSet = t.Iterable[DataTuple]
 
 
@@ -86,7 +87,7 @@ def sample_reviews_by_rating(reviews: pymongo.collection.Collection, rating: flo
     ])
 
 
-def review_to_datatuple(review: Review) -> tuple[Text, Category]:
+def review_to_datatuple(review: Review) -> DataTuple:
     """
     Return the label corresponding to the given review.
 
@@ -116,7 +117,7 @@ def review_to_datatuple(review: Review) -> tuple[Text, Category]:
         case _:
             category = "unknown"
 
-    return text, category
+    return DataTuple(text, category)
 
 
 def polar_dataset(collection: pymongo.collection.Collection, amount: int) -> t.Iterator[DataTuple]: