added gym-minigrid and fixed str | None

requirements.txt

@@ -15,3 +15,4 @@ torch
 ltlf2dfa
 minigrid
 click
+gym-minigrid

specless/dataset.py

@@ -44,7 +44,7 @@
 """
 import os
 from abc import abstractmethod
-from typing import Any, Callable, List
+from typing import Any, Callable, List, Optional
 
 import numpy as np
 import pandas as pd
@@ -74,7 +74,7 @@ def __getitem__(self, idx: int) -> Any:
     def apply(self, func: Callable[..., Any]) -> None:
         list(map(func, self.data))
 
-    def tolist(self, key: str | None = None):
+    def tolist(self, key: Optional[str] = None):
         if key is None:
             return list(map(lambda d: d.values.tolist(), self.data))
         else:
@@ -91,7 +91,7 @@ class CSVDataset(BaseDataset):
 
     def __init__(
         self,
-        filedir: str | None = None,
+        filedir: Optional[str] = None,
         filepaths: List[str] | None = None,
     ) -> None:
         """
@@ -106,7 +106,7 @@ def __init__(
         >> dataset = CSVDataset(filedir, filepaths=filepaths)
 
         Args:
-            filedir (str | None, optional): _description_. Defaults to None.
+            filedir (str, optional): _description_. Defaults to None.
         """
         if filedir is None and filepaths is None:
             raise Exception("Provide either filedir or filepaths")

specless/factory/tspadapter.py

@@ -237,7 +237,7 @@ def add_initial_state(self, initial_state: Tuple = None) -> None:
         self.obs_to_nodes[obs].append(node_id)
 
 
-class MiniGridSytemAndTSPAdapter(MiniGridSytemAndTSPAdapter):
+class MiniGridSytemAndTSPAdapterWithTPO(MiniGridSytemAndTSPAdapter):
     """Converts Transition System To a TSP Problem"""
 
     def __call__(

specless/inference.py

@@ -18,7 +18,7 @@
 from abc import ABCMeta, abstractmethod
 from collections import defaultdict
 from pathlib import Path
-from typing import Any, Callable, Dict, List, Tuple, Union
+from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
 import graphviz
 import networkx as nx
@@ -457,7 +457,7 @@ def __init__(
         self._learned_model_filepath: str
         self._initial_output_addon_name = "init_dfa"
         self._initial_model_filepath: str
-        self._output_base_filepath: str | None = None
+        self._output_base_filepath: Optional[str] = None
         self._output_directory: str = output_directory
         self._flexfringe_output_dir_popt_str: str = "o"
         Path(output_directory).mkdir(parents=True, exist_ok=True)

specless/inference/edsm.py

@@ -13,7 +13,7 @@
 import subprocess as sp
 import time
 from pathlib import Path
-from typing import Union
+from typing import Optional, Union
 
 import graphviz
 from IPython.display import Image, display
@@ -57,7 +57,7 @@ def __init__(
         self._learned_model_filepath: str
         self._initial_output_addon_name = "init_dfa"
         self._initial_model_filepath: str
-        self._output_base_filepath: str | None = None
+        self._output_base_filepath: Optional[str] = None
         self._output_directory: str = output_directory
         self._flexfringe_output_dir_popt_str: str = "o"
         Path(output_directory).mkdir(parents=True, exist_ok=True)