Skip to content

Commit 4da97d9

Browse files
committed
Added dataset accessor.
1 parent 07f204b commit 4da97d9

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

ads/dataset/dataset_accessor.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
from typing import Tuple
2+
import pandas as pd
3+
from ads.dataset import progress
4+
from ads.dataset.dataset import ADSDataset
5+
from ads.dataset.dataset_with_target import ADSDatasetWithTarget
6+
7+
@pd.api.extensions.register_dataframe_accessor("ads")
8+
class ADSDatasetAccessor:
9+
def __init__(self, pandas_obj):
10+
self._obj = pandas_obj

0 commit comments

Comments
 (0)