"""Checkpoint Initializer"""

from typing import Dict

import tensorflow as tf


class CheckpointInitializer:
    """Checkpoint Initializer"""

    def __init__(self, path_init_ckpt: str, assignment_map: Dict):
        self.path_init_ckpt = path_init_ckpt
        self.assignment_map = assignment_map

    def __call__(self):
        tf.train.init_from_checkpoint(self.path_init_ckpt, self.assignment_map)

9e552550c1ce7b9a5a00e57e75fa205a08fb4b3a
<mask token>


class CheckpointInitializer:
    <mask token>
    <mask token>

    def __call__(self):
        tf.train.init_from_checkpoint(self.path_init_ckpt, self.assignment_map)

<mask token>


class CheckpointInitializer:
    <mask token>

    def __init__(self, path_init_ckpt: str, assignment_map: Dict):
        self.path_init_ckpt = path_init_ckpt
        self.assignment_map = assignment_map

    def __call__(self):
        tf.train.init_from_checkpoint(self.path_init_ckpt, self.assignment_map)

<mask token>


class CheckpointInitializer:
    """Checkpoint Initializer"""

    def __init__(self, path_init_ckpt: str, assignment_map: Dict):
        self.path_init_ckpt = path_init_ckpt
        self.assignment_map = assignment_map

    def __call__(self):
        tf.train.init_from_checkpoint(self.path_init_ckpt, self.assignment_map)

<mask token>
from typing import Dict
import tensorflow as tf


class CheckpointInitializer:
    """Checkpoint Initializer"""

    def __init__(self, path_init_ckpt: str, assignment_map: Dict):
        self.path_init_ckpt = path_init_ckpt
        self.assignment_map = assignment_map

    def __call__(self):
        tf.train.init_from_checkpoint(self.path_init_ckpt, self.assignment_map)

"""Checkpoint Initializer"""

from typing import Dict

import tensorflow as tf


class CheckpointInitializer:
    """Checkpoint Initializer"""

    def __init__(self, path_init_ckpt: str, assignment_map: Dict):
        self.path_init_ckpt = path_init_ckpt
        self.assignment_map = assignment_map

    def __call__(self):
        tf.train.init_from_checkpoint(self.path_init_ckpt, self.assignment_map)
