Merge branch 'v9.5.0' into v2_main

thoughtspot · Aug 24, 2023 · 40bd943 · 40bd943
2 parents 9061a6d + daf31fe
commit 40bd943
Show file tree

Hide file tree

Showing 3 changed files with 40 additions and 3 deletions.
diff --git a/src/thoughtspot_tml/_scriptability.py b/src/thoughtspot_tml/_scriptability.py
diff --git a/src/thoughtspot_tml/_version.py b/src/thoughtspot_tml/_version.py
@@ -1 +1 @@
-__version__ = "2.0.6"
+__version__ = "2.0.7"
diff --git a/src/thoughtspot_tml/utils.py b/src/thoughtspot_tml/utils.py
@@ -289,8 +289,19 @@ def save(self, path: PathLike, *, info: Dict[str, Any] = None) -> None:
         with pathlib.Path(path).open(mode="w", encoding="UTF-8") as j:
             json.dump(data, j, indent=4)
 
-    def get_environment_guids(mapper, source, destination) -> Iterator[GUID, GUID]:
-        for key, environments in mapper._mapping.items():
+    def get_environment_guids(self, *, source: str, destination: str) -> Iterator[GUID, GUID]:
+        """
+        Iterate through all guid pairs between source and destination.
+
+        Parameters
+        ----------
+        source : str
+          name of the environment to fetch the source guid from
+
+        destination : str
+          name of the environment to fetch the mapped guid from
+        """
+        for key, environments in self._mapping.items():
             guid_src = environments[source]
             guid_dst = environments[destination]
             yield guid_src, guid_dst