Fix some typos (#141)

Author: jansel

helion/_compat.py

@@ -62,7 +62,7 @@ def torch_dtype_to_tl(torch_dtype: torch.dtype) -> object:
 
 @functools.cache
 def min_dot_size(
-    device: torch.device, lhs: torch.dtype, rhs: torch.torch.dtype
+    device: torch.device, lhs: torch.dtype, rhs: torch.dtype
 ) -> tuple[int, int, int]:
     if device.type != "cuda":
         # TODO(jansel): support non-cuda properly

helion/_compiler/compile_environment.py

@@ -44,7 +44,7 @@ class _TLS(Protocol):
 class CompileEnvironment:
     """
     Global state for the duration of a compilation.
-    There is a 1:1 mapping between this an BoundKernel,
+    There is a 1:1 mapping between this and a BoundKernel,
     and a single CompileEnvironment will be used for multiple Configs.
     No config or codegen specific state should be stored here.
     """

helion/_logging/_internal.py

@@ -19,9 +19,9 @@
 
 
 @dataclass
-class LogRegistery:
+class LogRegistry:
     """
-    This registery holds mappings for logging
+    This registry holds mappings for logging
     """
 
     # alias -> list of logs
@@ -31,25 +31,25 @@ class LogRegistery:
     log_levels: dict[str, int] = field(default_factory=dict)
 
 
-_LOG_REGISTERY = LogRegistery({"all": ["helion"]})
+_LOG_REGISTRY = LogRegistry({"all": ["helion"]})
 
 
 def parse_log_value(value: str) -> None:
     """
-    Given a string like "foo.bar,+baz.fizz" this function parses this string and converts
-    it to mapping of {"foo.bar": logging.INFO, "baz.fizz": logging.DEBUG}
-    and updates the registery with this mapping
+    Given a string like "foo.bar,+baz.fizz" this function parses the string and converts
+    it to a mapping of {"foo.bar": logging.INFO, "baz.fizz": logging.DEBUG}
+    and updates the registry with this mapping
     """
     entries = [e.strip() for e in value.split(",") if e.strip()]
     for entry in entries:
         log_level = logging.DEBUG if entry.startswith("+") else logging.INFO
         alias = entry.lstrip("+")
 
-        if alias in _LOG_REGISTERY.alias_map:
-            for log in _LOG_REGISTERY.alias_map[alias]:
-                _LOG_REGISTERY.log_levels[log] = log_level
+        if alias in _LOG_REGISTRY.alias_map:
+            for log in _LOG_REGISTRY.alias_map[alias]:
+                _LOG_REGISTRY.log_levels[log] = log_level
         else:
-            _LOG_REGISTERY.log_levels[alias] = log_level
+            _LOG_REGISTRY.log_levels[alias] = log_level
 
 
 def init_logs_from_string(value: str) -> None:
@@ -58,7 +58,7 @@ def init_logs_from_string(value: str) -> None:
     """
     parse_log_value(value)
 
-    for logger_name, level in _LOG_REGISTERY.log_levels.items():
+    for logger_name, level in _LOG_REGISTRY.log_levels.items():
         logger = logging.getLogger(logger_name)
         logger.setLevel(level)
 

helion/autotuner/base_search.py

@@ -466,7 +466,7 @@ def _mark_complete(self) -> bool:
         """
         Mark the precompile future as complete and kill the process if needed.
 
-        :return True if the precompilation was successful, False otherwise.
+        :return: True if the precompilation was successful, False otherwise.
         """
         self.end_time = time.time()
         process = self.process

helion/autotuner/logger.py

@@ -9,15 +9,15 @@
 
 class LambdaLogger:
     """
-    A self-contained logger that does not propagates to the root logger and
+    A self-contained logger that does not propagate to the root logger and
     prints each record to stderr in the form:
 
         [<elapsed>s] <message>
 
     where *elapsed* is the whole-second wall-clock time since the logger
     instance was created.
 
-    Takes lambas as arguments, which are called when the log is emitted.
+    Takes lambdas as arguments, which are called when the log is emitted.
     """
 
     _count: itertools.count[int] = itertools.count()

test/test_logging.py

@@ -17,11 +17,12 @@ def test_log_set(self):
         from helion._logging._internal import init_logs_from_string
 
         init_logs_from_string("foo.bar,+fuzz.baz")
-        self.assertTrue(
-            helion._logging._internal._LOG_REGISTERY.log_levels["foo.bar"], logging.INFO
+        self.assertEqual(
+            helion._logging._internal._LOG_REGISTRY.log_levels["foo.bar"],
+            logging.INFO,
         )
-        self.assertTrue(
-            helion._logging._internal._LOG_REGISTERY.log_levels["fuzz.baz"],
+        self.assertEqual(
+            helion._logging._internal._LOG_REGISTRY.log_levels["fuzz.baz"],
             logging.DEBUG,
         )