Verified Commit cfac2e28 authored by Hines, Jesse's avatar Hines, Jesse
Browse files

Fix running single partition

parent a546f993

raps/dataloaders/mit_supercloud/loader.py

+1 −1
Original line number Diff line number Diff line
@@ -298,7 +298,7 @@ def load_data(local_dataset_path, **kwargs): 
    ])



    # partition mode

    part = kwargs.get("partition", "").split("/")[-1].lower()

    part = (kwargs.get("partition") or "").split("/")[-1].lower()

    cpu_only = (part == "part-cpu")

    mixed = (part == "part-gpu")

raps/telemetry.py

+3 −2
Original line number Diff line number Diff line
@@ -77,11 +77,12 @@ class Telemetry: 


    def __init__(self, **kwargs):

        self.kwargs = kwargs

        self.system = kwargs.get('system')

        self.system = kwargs['system']

        self.config = kwargs.get('config')



        try:

            self.dataloader = importlib.import_module(f"raps.dataloaders.{self.system}", package=__package__)

            module = self.system.split("/")[0]

            self.dataloader = importlib.import_module(f"raps.dataloaders.{module}", package=__package__)

        except ImportError as e:

            print(f"WARNING: Failed to load dataloader: {e}")

            self.dataloader = None