def METHOD_NAME(
    self, sample_in_dim, sample_out_dim, sample_in_splits=1, sample_out_splits=1
):
    assert sample_in_dim is not None and sample_out_dim is not None
    if sample_in_splits is not None:
        assert sample_in_splits <= self.in_splits, f"{sample_in_splits}"
        assert (
            sample_in_dim // sample_in_splits > 0
        ), f"{sample_in_dim} // {sample_in_splits} <= 0"
    if sample_out_splits is not None:
        assert sample_out_splits <= self.out_splits, f"{sample_out_splits}"
        assert (
            sample_out_dim // sample_out_splits > 0
        ), f"{sample_out_dim} // {sample_out_splits} <= 0"
    self.sample_in_dim = sample_in_dim
    self.sample_out_dim = sample_out_dim
    self.sample_in_splits = sample_in_splits
    self.sample_out_splits = sample_out_splits
    self._sample_parameters()