org.mlflow.tracking.RunsPage Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mlflow-client Show documentation
There is a newer version: 2.21.0
package org.mlflow.tracking;

import java.util.Collections;
import java.util.List;
import java.util.ArrayList;
import java.util.Optional;
import org.mlflow.api.proto.Service.*;

public class RunsPage implements Page {

  private final String token;
  private final List runs;

  private final MlflowClient client;
  private final List experimentIds;
  private final String searchFilter;
  private final ViewType runViewType;
  private final List orderBy;
  private final int maxResults;

  /**
   * Creates a fixed size page of Runs.
   */
  RunsPage(List runs,
                  String token,
                  List experimentIds,
                  String searchFilter,
                  ViewType runViewType,
                  int maxResults,
                  List orderBy,
                  MlflowClient client) {
    this.runs = Collections.unmodifiableList(runs);
    this.token = token;
    this.experimentIds = experimentIds;
    this.searchFilter = searchFilter;
    this.runViewType = runViewType;
    this.orderBy = orderBy;
    this.maxResults = maxResults;
    this.client = client;
  }

  /**
   * @return The number of runs in the page.
   */
  public int getPageSize() {
    return this.runs.size();
  }

  /**
   * @return True if a token for the next page exists and isn't empty. Otherwise returns false.
   */
  public boolean hasNextPage() {
    return this.token != null && this.token != "";
  }

  /**
   * @return An optional with the token for the next page. 
   * Empty if the token doesn't exist or is empty.
   */
  public Optional getNextPageToken() {
    if (this.hasNextPage()) {
      return Optional.of(this.token);
    } else {
      return Optional.empty();
    }
  }

  /**
   * @return The next page of runs matching the search criteria. 
   * If there are no more pages, an {@link org.mlflow.tracking.EmptyPage} will be returned.
   */
  public Page getNextPage() {
    if (this.hasNextPage()) {
      return this.client.searchRuns(this.experimentIds,
                                    this.searchFilter,
                                    this.runViewType,
                                    this.maxResults,
                                    this.orderBy,
                                    this.token);
    } else {
      return new EmptyPage();
    }
  }

  /**
   * @return An iterable over the runs in this page.
   */
  public List getItems() {
    return runs;
  }

}