Run black formatter on code using:

* shopt -s globstar * black **.py
fire · Oct 8, 2020 · 686bd4d · 686bd4d
1 parent 662b2f0
commit 686bd4d
Show file tree

Hide file tree

Showing 36 changed files with 3,517 additions and 1,958 deletions.
diff --git a/datasets/__init__.py b/datasets/__init__.py
@@ -3,17 +3,40 @@ def get_character_names(args):
         """
         Put the name of subdirectory in retargeting/datasets/Mixamo as [[names of group A], [names of group B]]
         """
-        characters = [['Aj', 'BigVegas', 'Kaya', 'SportyGranny'],
-                      ['Malcolm_m', 'Remy_m', 'Maria_m', 'Jasper_m', 'Knight_m',
-                       'Liam_m', 'ParasiteLStarkie_m', 'Pearl_m', 'Michelle_m', 'LolaB_m',
-                       'Pumpkinhulk_m', 'Ortiz_m', 'Paladin_m', 'James_m', 'Joe_m',
-                       'Olivia_m', 'Yaku_m', 'Timmy_m', 'Racer_m', 'Abe_m']]
+        characters = [
+            ["Aj", "BigVegas", "Kaya", "SportyGranny"],
+            [
+                "Malcolm_m",
+                "Remy_m",
+                "Maria_m",
+                "Jasper_m",
+                "Knight_m",
+                "Liam_m",
+                "ParasiteLStarkie_m",
+                "Pearl_m",
+                "Michelle_m",
+                "LolaB_m",
+                "Pumpkinhulk_m",
+                "Ortiz_m",
+                "Paladin_m",
+                "James_m",
+                "Joe_m",
+                "Olivia_m",
+                "Yaku_m",
+                "Timmy_m",
+                "Racer_m",
+                "Abe_m",
+            ],
+        ]
 
     else:
         """
         To run evaluation successfully, number of characters in both groups must be the same. Repeat is okay.
         """
-        characters = [['BigVegas', 'BigVegas', 'BigVegas', 'BigVegas'],  ['Mousey_m', 'Goblin_m', 'Mremireh_m', 'Vampire_m']]
+        characters = [
+            ["BigVegas", "BigVegas", "BigVegas", "BigVegas"],
+            ["Mousey_m", "Goblin_m", "Mremireh_m", "Vampire_m"],
+        ]
         tmp = characters[1][args.eval_seq]
         characters[1][args.eval_seq] = characters[1][0]
         characters[1][0] = tmp
@@ -31,14 +54,14 @@ def create_dataset(args, character_names=None):
 
 
 def get_test_set():
-    with open('./datasets/Mixamo/test_list.txt', 'r') as file:
+    with open("./datasets/Mixamo/test_list.txt", "r") as file:
         list = file.readlines()
         list = [f[:-1] for f in list]
         return list
 
 
 def get_train_list():
-    with open('./datasets/Mixamo/train_list.txt', 'r') as file:
+    with open("./datasets/Mixamo/train_list.txt", "r") as file:
         list = file.readlines()
         list = [f[:-1] for f in list]
         return list