def METHOD_NAME(
    self, direction: LineageQueryDirectionEnum = LineageQueryDirectionEnum.ASCENDANTS
) -> List[Artifact]:
    """Use the lineage query to retrieve datasets that use this trial component.
    Args:
        direction (LineageQueryDirectionEnum, optional): The query direction.
    Returns:
        list of Artifacts: Artifacts representing a dataset.
    """
    query_filter = LineageFilter(
        entities=[LineageEntityEnum.ARTIFACT], sources=[LineageSourceEnum.DATASET]
    )
    query_result = LineageQuery(self.sagemaker_session).query(
        start_arns=[self.trial_component_arn],
        query_filter=query_filter,
        direction=direction,
        include_edges=False,
    )
    return [vertex.to_lineage_object() for vertex in query_result.vertices]