[qob] fix ServiceBackend.get_flags

I just plainly implemented this wrong the first time.
hail-is · Oct 12, 2022 · 669efef · 669efef
1 parent f12a5f0
commit 669efef
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/hail/python/hail/backend/service_backend.py b/hail/python/hail/backend/service_backend.py
@@ -727,7 +727,7 @@ def set_flags(self, **flags: str):
         self.flags.update(flags)
 
     def get_flags(self, *flags) -> Mapping[str, str]:
-        return frozendict(self.flags)
+        return {flag: self.flags[flag] for flag in flags}
 
     @property
     def requires_lowering(self):

diff --git a/hail/python/test/hail/test_context.py b/hail/python/test/hail/test_context.py
@@ -29,3 +29,7 @@ def test_top_level_functions_are_do_not_error(self):
 
     def test_tmpdir_runs(self):
         isinstance(hl.tmp_dir(), str)
+
+    def test_get_flags(self):
+        assert hl._get_flags() == set()
+        assert list(hl._get_flags('use_new_shuffle')) == 'use_new_shuffle'