[FIX] Fix Typo in common_write preventing to force cast to dict

wbenbihi · Aug 17, 2022 · 3529321 · 3529321
1 parent d051841
commit 3529321
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hourglass_tensorflow/utils/writers.py b/hourglass_tensorflow/utils/writers.py
@@ -40,7 +40,7 @@ def common_write(
         )
 
     output_obj = obj
-    if force_dict_struct or file_extension in [".yml", ".yaml", ".json"]:
+    if force_dict_struct or file_extension in ["yml", "yaml", "json"]:
         # We will cast any pydantic.BaseModel if
         #   - The force_dict_struct is set to True (pickle only)
         #   - The output file type require to write a dict