Commit 809d26cf authored by Brewer, Wes's avatar Brewer, Wes
Browse files

Output scheduled nodes in verbose mode for Lassen dataloader

parent 6b9bae3c

raps/dataloaders/lassen.py

+6 −4
Original line number Diff line number Diff line
@@ -123,10 +123,6 @@ def load_data_from_df(allocation_df, node_df, step_df, **kwargs): 
        ib_rx = 4 * node_data['ib_rx'].values[0] if node_data['ib_rx'].values.size > 0 else []



        net_tx, net_rx = generate_network_sequences(ib_tx, ib_rx, samples, lambda_poisson=0.3)

        if verbose:

            print('ib_tx, ib_rx, samples:', ib_tx, ib_rx, samples)

            print('tx:', net_tx)

            print('rx:', net_rx)



        if reschedule:  # Let the scheduler reschedule the jobs

            scheduled_nodes = None
@@ -137,6 +133,12 @@ def load_data_from_df(allocation_df, node_df, step_df, **kwargs): 
            if fastforward:

                time_offset -= fastforward



        if verbose:

            print('ib_tx, ib_rx, samples:', ib_tx, ib_rx, samples)

            print('tx:', net_tx)

            print('rx:', net_rx)

            print('scheduled_nodes:', nodes_required, scheduled_nodes)



        if time_offset >= 0:



            job_info = job_dict(nodes_required,