Skip to content

Commit 2b7583b

Browse files
fix: show inherited SDK methods in pydoc (#1707)
Co-authored-by: nayaknishant <[email protected]>
1 parent 9e1d796 commit 2b7583b

File tree

2 files changed

+1
-176
lines changed

2 files changed

+1
-176
lines changed

docs/aiplatform/services.rst

+1
Original file line numberDiff line numberDiff line change
@@ -4,3 +4,4 @@ Google Cloud Aiplatform SDK
44
.. automodule:: google.cloud.aiplatform
55
:members:
66
:show-inheritance:
7+
:inherited-members:

google/cloud/aiplatform/training_jobs.py

-176
Original file line numberDiff line numberDiff line change
@@ -5150,94 +5150,6 @@ class AutoMLForecastingTrainingJob(_ForecastingTrainingJob):
51505150
_training_task_definition = schema.training_job.definition.automl_forecasting
51515151
_supported_training_schemas = (schema.training_job.definition.automl_forecasting,)
51525152

5153-
def run(
5154-
self,
5155-
dataset: datasets.TimeSeriesDataset,
5156-
target_column: str,
5157-
time_column: str,
5158-
time_series_identifier_column: str,
5159-
unavailable_at_forecast_columns: List[str],
5160-
available_at_forecast_columns: List[str],
5161-
forecast_horizon: int,
5162-
data_granularity_unit: str,
5163-
data_granularity_count: int,
5164-
training_fraction_split: Optional[float] = None,
5165-
validation_fraction_split: Optional[float] = None,
5166-
test_fraction_split: Optional[float] = None,
5167-
predefined_split_column_name: Optional[str] = None,
5168-
timestamp_split_column_name: Optional[str] = None,
5169-
weight_column: Optional[str] = None,
5170-
time_series_attribute_columns: Optional[List[str]] = None,
5171-
context_window: Optional[int] = None,
5172-
export_evaluated_data_items: bool = False,
5173-
export_evaluated_data_items_bigquery_destination_uri: Optional[str] = None,
5174-
export_evaluated_data_items_override_destination: bool = False,
5175-
quantiles: Optional[List[float]] = None,
5176-
validation_options: Optional[str] = None,
5177-
budget_milli_node_hours: int = 1000,
5178-
model_display_name: Optional[str] = None,
5179-
model_labels: Optional[Dict[str, str]] = None,
5180-
model_id: Optional[str] = None,
5181-
parent_model: Optional[str] = None,
5182-
is_default_version: Optional[bool] = True,
5183-
model_version_aliases: Optional[Sequence[str]] = None,
5184-
model_version_description: Optional[str] = None,
5185-
additional_experiments: Optional[List[str]] = None,
5186-
hierarchy_group_columns: Optional[List[str]] = None,
5187-
hierarchy_group_total_weight: Optional[float] = None,
5188-
hierarchy_temporal_total_weight: Optional[float] = None,
5189-
hierarchy_group_temporal_total_weight: Optional[float] = None,
5190-
window_column: Optional[str] = None,
5191-
window_stride_length: Optional[int] = None,
5192-
window_max_count: Optional[int] = None,
5193-
holiday_regions: Optional[List[str]] = None,
5194-
sync: bool = True,
5195-
create_request_timeout: Optional[float] = None,
5196-
) -> models.Model:
5197-
return super().run(
5198-
dataset=dataset,
5199-
target_column=target_column,
5200-
time_column=time_column,
5201-
time_series_identifier_column=time_series_identifier_column,
5202-
unavailable_at_forecast_columns=unavailable_at_forecast_columns,
5203-
available_at_forecast_columns=available_at_forecast_columns,
5204-
forecast_horizon=forecast_horizon,
5205-
data_granularity_unit=data_granularity_unit,
5206-
data_granularity_count=data_granularity_count,
5207-
training_fraction_split=training_fraction_split,
5208-
validation_fraction_split=validation_fraction_split,
5209-
test_fraction_split=test_fraction_split,
5210-
predefined_split_column_name=predefined_split_column_name,
5211-
timestamp_split_column_name=timestamp_split_column_name,
5212-
weight_column=weight_column,
5213-
time_series_attribute_columns=time_series_attribute_columns,
5214-
context_window=context_window,
5215-
budget_milli_node_hours=budget_milli_node_hours,
5216-
export_evaluated_data_items=export_evaluated_data_items,
5217-
export_evaluated_data_items_bigquery_destination_uri=export_evaluated_data_items_bigquery_destination_uri,
5218-
export_evaluated_data_items_override_destination=export_evaluated_data_items_override_destination,
5219-
quantiles=quantiles,
5220-
validation_options=validation_options,
5221-
model_display_name=model_display_name,
5222-
model_labels=model_labels,
5223-
model_id=model_id,
5224-
parent_model=parent_model,
5225-
is_default_version=is_default_version,
5226-
model_version_aliases=model_version_aliases,
5227-
model_version_description=model_version_description,
5228-
additional_experiments=additional_experiments,
5229-
hierarchy_group_columns=hierarchy_group_columns,
5230-
hierarchy_group_total_weight=hierarchy_group_total_weight,
5231-
hierarchy_temporal_total_weight=hierarchy_temporal_total_weight,
5232-
hierarchy_group_temporal_total_weight=hierarchy_group_temporal_total_weight,
5233-
window_column=window_column,
5234-
window_stride_length=window_stride_length,
5235-
window_max_count=window_max_count,
5236-
holiday_regions=holiday_regions,
5237-
sync=sync,
5238-
create_request_timeout=create_request_timeout,
5239-
)
5240-
52415153

52425154
class SequenceToSequencePlusForecastingTrainingJob(_ForecastingTrainingJob):
52435155
_model_type = "Seq2Seq"
@@ -5246,94 +5158,6 @@ class SequenceToSequencePlusForecastingTrainingJob(_ForecastingTrainingJob):
52465158
schema.training_job.definition.seq2seq_plus_forecasting,
52475159
)
52485160

5249-
def run(
5250-
self,
5251-
dataset: datasets.TimeSeriesDataset,
5252-
target_column: str,
5253-
time_column: str,
5254-
time_series_identifier_column: str,
5255-
unavailable_at_forecast_columns: List[str],
5256-
available_at_forecast_columns: List[str],
5257-
forecast_horizon: int,
5258-
data_granularity_unit: str,
5259-
data_granularity_count: int,
5260-
training_fraction_split: Optional[float] = None,
5261-
validation_fraction_split: Optional[float] = None,
5262-
test_fraction_split: Optional[float] = None,
5263-
predefined_split_column_name: Optional[str] = None,
5264-
timestamp_split_column_name: Optional[str] = None,
5265-
weight_column: Optional[str] = None,
5266-
time_series_attribute_columns: Optional[List[str]] = None,
5267-
context_window: Optional[int] = None,
5268-
export_evaluated_data_items: bool = False,
5269-
export_evaluated_data_items_bigquery_destination_uri: Optional[str] = None,
5270-
export_evaluated_data_items_override_destination: bool = False,
5271-
quantiles: Optional[List[float]] = None,
5272-
validation_options: Optional[str] = None,
5273-
budget_milli_node_hours: int = 1000,
5274-
model_display_name: Optional[str] = None,
5275-
model_labels: Optional[Dict[str, str]] = None,
5276-
model_id: Optional[str] = None,
5277-
parent_model: Optional[str] = None,
5278-
is_default_version: Optional[bool] = True,
5279-
model_version_aliases: Optional[Sequence[str]] = None,
5280-
model_version_description: Optional[str] = None,
5281-
additional_experiments: Optional[List[str]] = None,
5282-
hierarchy_group_columns: Optional[List[str]] = None,
5283-
hierarchy_group_total_weight: Optional[float] = None,
5284-
hierarchy_temporal_total_weight: Optional[float] = None,
5285-
hierarchy_group_temporal_total_weight: Optional[float] = None,
5286-
window_column: Optional[str] = None,
5287-
window_stride_length: Optional[int] = None,
5288-
window_max_count: Optional[int] = None,
5289-
holiday_regions: Optional[List[str]] = None,
5290-
sync: bool = True,
5291-
create_request_timeout: Optional[float] = None,
5292-
) -> models.Model:
5293-
return super().run(
5294-
dataset=dataset,
5295-
target_column=target_column,
5296-
time_column=time_column,
5297-
time_series_identifier_column=time_series_identifier_column,
5298-
unavailable_at_forecast_columns=unavailable_at_forecast_columns,
5299-
available_at_forecast_columns=available_at_forecast_columns,
5300-
forecast_horizon=forecast_horizon,
5301-
data_granularity_unit=data_granularity_unit,
5302-
data_granularity_count=data_granularity_count,
5303-
training_fraction_split=training_fraction_split,
5304-
validation_fraction_split=validation_fraction_split,
5305-
test_fraction_split=test_fraction_split,
5306-
predefined_split_column_name=predefined_split_column_name,
5307-
timestamp_split_column_name=timestamp_split_column_name,
5308-
model_id=model_id,
5309-
parent_model=parent_model,
5310-
is_default_version=is_default_version,
5311-
model_version_aliases=model_version_aliases,
5312-
model_version_description=model_version_description,
5313-
weight_column=weight_column,
5314-
time_series_attribute_columns=time_series_attribute_columns,
5315-
context_window=context_window,
5316-
budget_milli_node_hours=budget_milli_node_hours,
5317-
export_evaluated_data_items=export_evaluated_data_items,
5318-
export_evaluated_data_items_bigquery_destination_uri=export_evaluated_data_items_bigquery_destination_uri,
5319-
export_evaluated_data_items_override_destination=export_evaluated_data_items_override_destination,
5320-
quantiles=quantiles,
5321-
validation_options=validation_options,
5322-
model_display_name=model_display_name,
5323-
model_labels=model_labels,
5324-
additional_experiments=additional_experiments,
5325-
hierarchy_group_columns=hierarchy_group_columns,
5326-
hierarchy_group_total_weight=hierarchy_group_total_weight,
5327-
hierarchy_temporal_total_weight=hierarchy_temporal_total_weight,
5328-
hierarchy_group_temporal_total_weight=hierarchy_group_temporal_total_weight,
5329-
window_column=window_column,
5330-
window_stride_length=window_stride_length,
5331-
window_max_count=window_max_count,
5332-
holiday_regions=holiday_regions,
5333-
sync=sync,
5334-
create_request_timeout=create_request_timeout,
5335-
)
5336-
53375161

53385162
class AutoMLImageTrainingJob(_TrainingJob):
53395163
_supported_training_schemas = (

0 commit comments

Comments
 (0)