Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 11 additions & 12 deletions toolset_distributions.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
all_dists = list_distributions()
"""

from typing import Dict, List, Optional
from typing import Any, Dict, List, Optional
import random
from toolsets import validate_toolset

Expand Down Expand Up @@ -220,26 +220,25 @@
}


def get_distribution(name: str) -> Optional[Dict[str, any]]:
def get_distribution(name: str) -> Optional[Dict[str, Any]]:
"""
Get a toolset distribution by name.

Args:
name (str): Name of the distribution
name: Name of the distribution

Returns:
Dict: Distribution definition with description and toolsets
None: If distribution not found
Distribution definition with description and toolsets, or None if not found
"""
return DISTRIBUTIONS.get(name)


def list_distributions() -> Dict[str, Dict]:
def list_distributions() -> Dict[str, Dict[str, Any]]:
"""
List all available distributions.

Returns:
Dict: All distribution definitions
Copy of all distribution definitions
"""
return DISTRIBUTIONS.copy()

Expand All @@ -252,10 +251,10 @@ def sample_toolsets_from_distribution(distribution_name: str) -> List[str]:
This allows multiple toolsets to be active simultaneously.

Args:
distribution_name (str): Name of the distribution to sample from
distribution_name: Name of the distribution to sample from

Returns:
List[str]: List of sampled toolset names
List of sampled toolset names

Raises:
ValueError: If distribution name is not found
Expand Down Expand Up @@ -293,10 +292,10 @@ def validate_distribution(distribution_name: str) -> bool:
Check if a distribution name is valid.

Args:
distribution_name (str): Distribution name to validate
distribution_name: Distribution name to validate

Returns:
bool: True if valid, False otherwise
True if valid, False otherwise
"""
return distribution_name in DISTRIBUTIONS

Expand All @@ -306,7 +305,7 @@ def print_distribution_info(distribution_name: str) -> None:
Print detailed information about a distribution.

Args:
distribution_name (str): Distribution name
distribution_name: Distribution name to display
"""
dist = get_distribution(distribution_name)
if not dist:
Expand Down