From 9be0c847154b7a20cfd3f51fc0c366099c007f5c Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Thu, 7 Jul 2022 15:33:53 -0700
Subject: [PATCH] Change set -> dict w/ None keys for dataset split synonym
 search, so always consistent if more than 1 exists. Fix #1224

---
 timm/data/dataset_factory.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/timm/data/dataset_factory.py b/timm/data/dataset_factory.py
index 194a597e..d0ac30b1 100644
--- a/timm/data/dataset_factory.py
+++ b/timm/data/dataset_factory.py
@@ -26,8 +26,8 @@ _TORCH_BASIC_DS = dict(
     kmnist=KMNIST,
     fashion_mnist=FashionMNIST,
 )
-_TRAIN_SYNONYM = {'train', 'training'}
-_EVAL_SYNONYM = {'val', 'valid', 'validation', 'eval', 'evaluation'}
+_TRAIN_SYNONYM = dict(train=None, training=None)
+_EVAL_SYNONYM = dict(val=None, valid=None, validation=None, eval=None, evaluation=None)
 
 
 def _search_split(root, split):