Find a saved dataset in the registry by provided name and create a retrieval job to pull whole dataset from storage (offline store). If dataset couldn't be found by provided name SavedDatasetNotFound exception will be raised. Data will be retrieved from globally co
(self, name: str)
| 1938 | return dataset |
| 1939 | |
| 1940 | def get_saved_dataset(self, name: str) -> SavedDataset: |
| 1941 | """ |
| 1942 | Find a saved dataset in the registry by provided name and |
| 1943 | create a retrieval job to pull whole dataset from storage (offline store). |
| 1944 | |
| 1945 | If dataset couldn't be found by provided name SavedDatasetNotFound exception will be raised. |
| 1946 | |
| 1947 | Data will be retrieved from globally configured offline store. |
| 1948 | |
| 1949 | Returns: |
| 1950 | SavedDataset with RetrievalJob attached |
| 1951 | |
| 1952 | Raises: |
| 1953 | SavedDatasetNotFound |
| 1954 | """ |
| 1955 | if not flags_helper.is_test(): |
| 1956 | warnings.warn( |
| 1957 | "Retrieving datasets is an experimental feature. " |
| 1958 | "This API is unstable and it could and most probably will be changed in the future. " |
| 1959 | "We do not guarantee that future changes will maintain backward compatibility.", |
| 1960 | RuntimeWarning, |
| 1961 | ) |
| 1962 | |
| 1963 | dataset = self.registry.get_saved_dataset(name, self.project) |
| 1964 | provider = self._get_provider() |
| 1965 | |
| 1966 | retrieval_job = provider.retrieve_saved_dataset( |
| 1967 | config=self.config, dataset=dataset |
| 1968 | ) |
| 1969 | return dataset.with_retrieval_job(retrieval_job) |
| 1970 | |
| 1971 | def _materialize_odfv( |
| 1972 | self, |