{"label": 1}, {"label": 2}, {"label": 0}, {"label": 0}, {"label": 0}, ] assert ds[1].compute(label_name=True) == {"label": "blue"} assert ds[1].compute() == {"label": 2} assert ds[1:3].compute(label_name=True).tolist() == [ {"label": "blue"}, {"label": "red"}, ] assert ds[1:3].compute().tolist() == [{"label": 2}, {"label": 0}] @pytest.mark.skipif(not minio_creds_exist(), reason="requires minio credentials") def test_minio_endpoint(): token = { "aws_access_key_id": os.getenv("ACTIVELOOP_MINIO_KEY"), "aws_secret_access_key": os.getenv("ACTIVELOOP_MINIO_SECRET_ACCESS_KEY"), "endpoint_url": "https://play.min.io:9000", "region": "us-east-1", } schema = {"abc": Tensor((100, 100, 3))} ds = Dataset( "s3://bucket/random_dataset", token=token, shape=(10,), schema=schema, mode="w" ) for i in range(10): ds["abc", i] = i * np.ones((100, 100, 3))
assert isinstance(ex, ClassLabelValueError) try: ds[0:4]["label/b"] = np.array([0, 1, 2, 3, 7]) except Exception as ex: assert isinstance(ex, ClassLabelValueError) try: ds["label", 4] = "name4" except Exception as ex: assert isinstance(ex, ClassLabelValueError) try: ds[0]["label/b"] = ["name"] except Exception as ex: assert isinstance(ex, ValueError) @pytest.mark.skipif(not minio_creds_exist(), reason="requires minio credentials") def test_minio_endpoint(): token = { "aws_access_key_id": os.getenv("ACTIVELOOP_MINIO_KEY"), "aws_secret_access_key": os.getenv("ACTIVELOOP_MINIO_SECRET_ACCESS_KEY"), "endpoint_url": "https://play.min.io:9000", "region": "us-east-1", } schema = {"abc": Tensor((100, 100, 3))} ds = Dataset("s3://bucket/random_dataset", token=token, shape=(10, ), schema=schema,