Add and run isort

.github/workflows/ci.yml

@@ -23,6 +23,9 @@ jobs:
       - name: run black
         run: black --check .
 
+      - name: run isort
+        run: isort .
+
       - name: run pylint for mdp folder
         run: pylint src/behavior_generation_lecture_python/mdp --errors-only
 

pyproject.toml

@@ -26,7 +26,8 @@ dev = [
   "pytest",
   "pytest-cov>=3.0.0",
   "pylint",
-  "mypy"
+  "mypy",
+  "isort"
 ]
 docs = [
   "mkdocs-material",
@@ -45,3 +46,6 @@ docs = [
 [build-system]
 requires = ["setuptools>=64.0.0", "wheel", "pip>=21.3.0"]
 build-backend = "setuptools.build_meta"
+
+[tool.isort]
+profile = "black"

scripts/run_a_star.py

@@ -1,6 +1,6 @@
 import numpy as np
 
-from behavior_generation_lecture_python.graph_search.a_star import Node, Graph
+from behavior_generation_lecture_python.graph_search.a_star import Graph, Node
 
 
 def main():

src/behavior_generation_lecture_python/graph_search/a_star.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 
+from typing import Dict, List, Set
+
 import matplotlib.pyplot as plt
 import numpy as np
 
-from typing import Dict, List, Set
-
 
 class Node:
     def __init__(

src/behavior_generation_lecture_python/utils/grid_plotting.py

@@ -1,4 +1,5 @@
 from collections import defaultdict
+
 import matplotlib.pyplot as plt
 import numpy as np
 

tests/test_a_star.py

@@ -1,7 +1,7 @@
 import matplotlib
 import numpy as np
 
-from behavior_generation_lecture_python.graph_search.a_star import Node, Graph
+from behavior_generation_lecture_python.graph_search.a_star import Graph, Node
 
 
 def test_example_graph():

tests/test_mdp.py

@@ -5,13 +5,13 @@
     MDP,
     SIMPLE_MDP_DICT,
     GridMDP,
+    best_action_from_q_table,
     derive_policy,
     expected_utility_of_action,
-    value_iteration,
-    best_action_from_q_table,
-    random_action,
     greedy_value_estimate_for_state,
     q_learning,
+    random_action,
+    value_iteration,
 )
 
 

tests/utils/test_grid_plotting.py

@@ -1,14 +1,14 @@
 import matplotlib
 
-from behavior_generation_lecture_python.utils.grid_plotting import (
-    make_plot_grid_step_function,
-    make_plot_policy_step_function,
-)
 from behavior_generation_lecture_python.mdp.mdp import (
     GRID_MDP_DICT,
     GridMDP,
     derive_policy,
 )
+from behavior_generation_lecture_python.utils.grid_plotting import (
+    make_plot_grid_step_function,
+    make_plot_policy_step_function,
+)
 
 TRUE_UTILITY_GRID_MDP = {
     (0, 0): 0.705,