code: def get_server_data(args, machine_id):
      """Get data from data_path/dataset/part_machine_id
  
      Return: glocal2local,
              entity_emb,
              entity_state,
              relation_emb,
              relation_emb_state
      """
      g2l, dataset = get_server_partition_dataset(
          args.data_path, args.dataset, machine_id
      )
  
      # Note that the dataset doesn't ccontain the triple
      print("n_entities: " + str(dataset.n_entities))
      print("n_relations: " + str(dataset.n_relations))
  
      args.soft_rel_part = False
      args.strict_rel_part = False
  
      model = load_model(args, dataset.n_entities, dataset.n_relations)
  
      return (
          g2l,
          model.entity_emb.emb,
          model.entity_emb.state_sum,
          model.relation_emb.emb,
          model.relation_emb.state_sum,
      )
  
imports: import os
  import argparse
  import time
  import socket
  import sys
  import dgl
  from dgl.contrib import KVServer
  import torch
  from .train_pytorch import load_model
  from .dataloader import get_server_partition_dataset
label_desc: Use image node_id 1 for calling a global function with example usage: get_server_data(args, machine_id) and returns: g2l,  model,  model,  model,  model