import tensorflow as tf

from .. import var_init



def embeddings(*, id_space_size, embedding_size):
  return tf.Variable(var_init.normal([id_space_size, embedding_size]))

import tensorflow as tf

from ..variable import variable



def embeddings(*, id_space_size, embedding_size):
  return variable([id_space_size, embedding_size])

Use variable module instead of var_init
Use variable module instead of var_init

  import tensorflow as tf
  
- from .. import var_init
+ from ..variable import variable
  
  
  
  def embeddings(*, id_space_size, embedding_size):
-   return tf.Variable(var_init.normal([id_space_size, embedding_size]))
+   return variable([id_space_size, embedding_size])
  
Use variable module instead of var_init
## Code Before:
import tensorflow as tf

from .. import var_init



def embeddings(*, id_space_size, embedding_size):
  return tf.Variable(var_init.normal([id_space_size, embedding_size]))

## Instruction:
Use variable module instead of var_init
## Code After:
import tensorflow as tf

from ..variable import variable



def embeddings(*, id_space_size, embedding_size):
  return variable([id_space_size, embedding_size])
