fixed indexable field via cast

build_map.py

@@ -143,16 +143,14 @@ def upload_dataset_to_atlas(dataset_dict,
 
 
     # return longest string field from 5 samples
+    head = dataset_dict["head"]
     if indexed_field is None:
         longest_length = 0
         for field in string_fields:
             length = 0
-            for i in range(len(dataset_dict["head"])):
-                ex = dataset_dict["head"].take([i])
-                print(ex)
-                if ex[field]:
-                    print(ex[field])
-                    length += len(ex[field].split())
+            for i in range(len(head)):
+                if head[field][i]:
+                    length += len(str(head[field][i]).split())
             if length > longest_length:
                 longest_length = length
                 indexed_field = field
@@ -262,4 +260,5 @@ if __name__ == "__main__":
     project_name = "huggingface_auto_upload_test-dolly-15k"
 
     dataset_dict = load_dataset_and_metadata(dataset_name)
-    print(upload_dataset_to_atlas(dataset_dict, project_name=project_name))
+    api_token = "ODdPKqJHYci4Gq4jnCC5-VR0L-rnIdfIy-6djgC4CTPCJ"
+    print(upload_dataset_to_atlas(dataset_dict, api_token, project_name=project_name))