cosmetic enhancements

Author: erindru

sqlmesh/core/snapshot/definition.py

@@ -1487,9 +1487,9 @@ class SnapshotTableCleanupTask(PydanticModel):
     dev_table_only: bool
 
 
-SnapshotIdLike = t.Union[SnapshotId, SnapshotTableInfo, Snapshot]
+SnapshotIdLike = t.Union[SnapshotId, SnapshotTableInfo, MinimalSnapshot, Snapshot]
 SnapshotInfoLike = t.Union[SnapshotTableInfo, Snapshot]
-SnapshotNameVersionLike = t.Union[SnapshotNameVersion, SnapshotTableInfo, Snapshot]
+SnapshotNameVersionLike = t.Union[SnapshotNameVersion, SnapshotTableInfo, MinimalSnapshot, Snapshot]
 
 
 class DeployabilityIndex(PydanticModel, frozen=True):

sqlmesh/core/state_sync/base.py

@@ -105,10 +105,10 @@ def get_snapshots_by_names(
         current_ts: t.Optional[int] = None,
         exclude_expired: bool = True,
     ) -> t.Set[MinimalSnapshot]:
-        """Return the snapshot id's for all versions of the specified snapshot names.
+        """Return the snapshot records for all versions of the specified snapshot names.
 
         Args:
-            snapshot_names: Iterable of snapshot names to fetch all snapshot for
+            snapshot_names: Iterable of snapshot names to fetch all snapshot records for
             current_ts: Sets the current time for identifying which snapshots have expired so they can be excluded (only relevant if :exclude_expired=True)
             exclude_expired: Whether or not to return the snapshot id's of expired snapshots in the result
 

sqlmesh/core/state_sync/db/snapshot.py

@@ -313,15 +313,15 @@ def get_snapshots_by_names(
         current_ts: t.Optional[int] = None,
         exclude_expired: bool = True,
     ) -> t.Set[MinimalSnapshot]:
-        """Return the snapshot id's for all versions of the specified snapshot names.
+        """Return the snapshot records for all versions of the specified snapshot names.
 
         Args:
-            snapshot_names: Iterable of snapshot names to fetch all snapshot id's for
+            snapshot_names: Iterable of snapshot names to fetch all snapshot records for
             current_ts: Sets the current time for identifying which snapshots have expired so they can be excluded (only relevant if :exclude_expired=True)
             exclude_expired: Whether or not to return the snapshot id's of expired snapshots in the result
 
         Returns:
-            A dictionary mapping snapshot names to a list of relevant snapshot id's
+            A set containing all the matched snapshot records. To fetch full snapshots, pass it into StateSync.get_snapshots()
         """
         if not snapshot_names:
             return set()