the-algorithm/src/scala/com/twitter/simclusters_v2/scio/multi_type_graph/multi_type_graph_sims/cosine-similarity-scio-adhoc.d6w
twitter-team ef4c5eb65e Twitter Recommendation Algorithm
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
2023-03-31 17:36:31 -05:00

34 lines
1.1 KiB
Plaintext

class Profile(Struct):
project = Required(String)
date = Required(String)
environment = Default(String, 'dev')
machine = Default(String, 'n2d-highmem-16')
job = Job(
name='multi-type-graph-cosine-similarity-scio-adhoc-app',
project='{{profile.project}}',
staging_bucket='{{profile.project}}',
service_account='{{profile.user_name}}-shdw@twttr-dp-svc-accounts.iam.gserviceaccount.com',
region='us-central1',
worker_config=WorkerConfig(
num_workers=2,
worker_machine_type='{{profile.machine}}',
worker_disk_type=WorkerDiskType('HDD'),
),
extra_args={
"environment": '{{profile.environment}}',
"date": Quote('{{profile.date}}'),
},
deployment_config=BatchDeploymentConfig(
role='{{profile.user_name}}',
build_target='src/scala/com/twitter/simclusters_v2/scio/multi_type_graph/multi_type_graph_sims:multi-type-graph-cosine-similarity-scio-adhoc-app',
gcp_deployment_credentials='/var/lib/tss/keys/{{profile.user_name}}/cloud/gcp/dp/shadow.json',
statebird_config=StatebirdConfig(
batch_width='PT1H',
first_time='{{profile.date}}'
)
)
)
jobs=[job]