for some reason colab doesn't like NamedTuple

Signed-off-by: Niv Sardi <[email protected]>

python/entity.py

@@ -4,14 +4,10 @@ from typing import NamedTuple
 
 from common import defaults
 
-def update(d, i):
-    d.update(i)
-    return d
-
 def read_entities(fn = defaults.MAIN_CSV_PATH):
     with open(fn, newline='') as csvfile:
         reader = csv.DictReader(csvfile)
-        bcos = { d['bco']:update(d, {'id': i}) for i, d in enumerate(reader)}
+        bcos = { d['bco']:d for  d in reader}
     return bcos
 
 class Entity(NamedTuple):

python/write_data.py

@@ -5,7 +5,7 @@ import argparse
 from common import defaults
 
 def gen_data_yaml(bcos):
-    names = [f"{d.name}" for d in bcos.values()]
+    names = [f"{d['name']}" for d in bcos.values()]
     return f'''
 train: ../data/squares
 val: ../data/squares