update format

projects/self_notes/README.md

@@ -17,7 +17,7 @@ the ball is inside the basket.
 Frank is with John.
 Daniel has the basket.
 Mary is at the museum.
-> 
+>
 > Q: Where is the basket?
 A: the basket is at the museum.
 
@@ -50,4 +50,4 @@ If you use our code in your own work, please cite with the following BibTex entr
     booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
     year={2023},
 }
-```
+```

projects/self_notes/toy_story/constants.py

@@ -7,9 +7,8 @@
 
 from typing import List, Type
 
-from relation import IsAtEdge, IsWithEdge, HasEdge, IsInsideEdge, create_rules
 from graph import Edge
-
+from relation import HasEdge, IsAtEdge, IsInsideEdge, IsWithEdge, create_rules
 
 PEOPLE = ["Alice", "Bob", "Charlie", "Daniel", "Frank", "Mary", "John", "Sandra"]
 PLACES = [

projects/self_notes/toy_story/main.py

@@ -5,17 +5,17 @@
 LICENSE file in the root directory of this source tree.
 """
 
-import json
-import random
 import argparse
+import json
 import os
+import random
 from os import path
 from typing import Any, Dict, List
-from tqdm import tqdm
-import numpy as np
 
-from world import World
+import numpy as np
 from constants import MAX_SAMPLE_TRIAL
+from tqdm import tqdm
+from world import World
 
 
 def save_json(data: List, file_name: str):

projects/self_notes/toy_story/relation.py

@@ -7,14 +7,14 @@
 
 from typing import List, Tuple
 
+from graph import Edge, Node
 from rules import (
     CheckRule,
     InferenceRule,
-    UniqueRule,
     SymmetryInference,
     TriangleInference,
+    UniqueRule,
 )
-from graph import Edge, Node
 
 
 class IsAtEdge(Edge):

projects/self_notes/toy_story/world.py

@@ -7,23 +7,23 @@
 
 import logging
 import random
-import numpy as np
 from typing import Any, Dict, List, Optional, Tuple, Type
 
-from graph import Edge, Graph
-from relation import IsAtEdge, IsWithEdge, HasEdge, IsInsideEdge, create_rules
-from rules import TriangleInference
+import numpy as np
 from constants import (
-    PEOPLE,
-    SMALL_ITEMS,
     CONTAINER_ITEMS,
-    PLACES,
-    RELATIONS,
-    QUESTION_RELATIONS,
-    MAX_OBS_TRIAL,
     MAX_NUM_SUPPORT,
+    MAX_OBS_TRIAL,
     MAX_QUESTION_TRIAL,
+    PEOPLE,
+    PLACES,
+    QUESTION_RELATIONS,
+    RELATIONS,
+    SMALL_ITEMS,
 )
+from graph import Edge, Graph
+from relation import HasEdge, IsAtEdge, IsInsideEdge, IsWithEdge, create_rules
+from rules import TriangleInference
 
 
 class World: