"""Some syft imports..."""

from syft import dp
from syft import core
from syft import spdz
from syft.core.frameworks.torch import _SyftTensor

from syft.core.frameworks.torch import TorchHook
from syft.core.frameworks.torch import (
    _LocalTensor,
    _PointerTensor,
    _LogTensor,
    _FixedPrecisionTensor,
    _PlusIsMinusTensor,
    _GeneralizedPointerTensor,
    _SPDZTensor,
    _SNNTensor,
)
from syft.core.workers import VirtualWorker, SocketWorker
from syft.core.frameworks.numpy import array

from torch.autograd import Variable, Variable as Var
from torch.nn import Parameter

__all__ = [
    "core",
    "spdz",
    "dp",
    "TorchHook",
    "_LocalTensor",
    "_PointerTensor",
    "_LogTensor",
    "_FixedPrecisionTensor",
    "_PlusIsMinusTensor",
    "_GeneralizedPointerTensor",
    "_SPDZTensor",
    "_SNNTensor",
    "VirtualWorker",
    "SocketWorker",
    "array",
    "Variable",
    "Var",
    "Parameter",
]


import syft
import torch

for f in dir(torch):
    if "_" not in f:
        setattr(syft, f, getattr(torch, f))

setattr(syft, "deser", _SyftTensor.deser)


# TODO: figure out how to let this be hooked here so that it happens
# automatically when you import syft. Right now it breaks if you accidentally
# hook again or if you need to hook it with a special local_worker
# (such as SocketWorker)
# hook = TorchHook(verbose=False)

34bcc287432fa5716805e79ca81b91f2c605f418
<mask token>

<mask token>
for f in dir(torch):
    if '_' not in f:
        setattr(syft, f, getattr(torch, f))
setattr(syft, 'deser', _SyftTensor.deser)

<mask token>
__all__ = ['core', 'spdz', 'dp', 'TorchHook', '_LocalTensor',
    '_PointerTensor', '_LogTensor', '_FixedPrecisionTensor',
    '_PlusIsMinusTensor', '_GeneralizedPointerTensor', '_SPDZTensor',
    '_SNNTensor', 'VirtualWorker', 'SocketWorker', 'array', 'Variable',
    'Var', 'Parameter']
<mask token>
for f in dir(torch):
    if '_' not in f:
        setattr(syft, f, getattr(torch, f))
setattr(syft, 'deser', _SyftTensor.deser)

<mask token>
from syft import dp
from syft import core
from syft import spdz
from syft.core.frameworks.torch import _SyftTensor
from syft.core.frameworks.torch import TorchHook
from syft.core.frameworks.torch import _LocalTensor, _PointerTensor, _LogTensor, _FixedPrecisionTensor, _PlusIsMinusTensor, _GeneralizedPointerTensor, _SPDZTensor, _SNNTensor
from syft.core.workers import VirtualWorker, SocketWorker
from syft.core.frameworks.numpy import array
from torch.autograd import Variable, Variable as Var
from torch.nn import Parameter
__all__ = ['core', 'spdz', 'dp', 'TorchHook', '_LocalTensor',
    '_PointerTensor', '_LogTensor', '_FixedPrecisionTensor',
    '_PlusIsMinusTensor', '_GeneralizedPointerTensor', '_SPDZTensor',
    '_SNNTensor', 'VirtualWorker', 'SocketWorker', 'array', 'Variable',
    'Var', 'Parameter']
import syft
import torch
for f in dir(torch):
    if '_' not in f:
        setattr(syft, f, getattr(torch, f))
setattr(syft, 'deser', _SyftTensor.deser)

"""Some syft imports..."""

from syft import dp
from syft import core
from syft import spdz
from syft.core.frameworks.torch import _SyftTensor

from syft.core.frameworks.torch import TorchHook
from syft.core.frameworks.torch import (
    _LocalTensor,
    _PointerTensor,
    _LogTensor,
    _FixedPrecisionTensor,
    _PlusIsMinusTensor,
    _GeneralizedPointerTensor,
    _SPDZTensor,
    _SNNTensor,
)
from syft.core.workers import VirtualWorker, SocketWorker
from syft.core.frameworks.numpy import array

from torch.autograd import Variable, Variable as Var
from torch.nn import Parameter

__all__ = [
    "core",
    "spdz",
    "dp",
    "TorchHook",
    "_LocalTensor",
    "_PointerTensor",
    "_LogTensor",
    "_FixedPrecisionTensor",
    "_PlusIsMinusTensor",
    "_GeneralizedPointerTensor",
    "_SPDZTensor",
    "_SNNTensor",
    "VirtualWorker",
    "SocketWorker",
    "array",
    "Variable",
    "Var",
    "Parameter",
]


import syft
import torch

for f in dir(torch):
    if "_" not in f:
        setattr(syft, f, getattr(torch, f))

setattr(syft, "deser", _SyftTensor.deser)


# TODO: figure out how to let this be hooked here so that it happens
# automatically when you import syft. Right now it breaks if you accidentally
# hook again or if you need to hook it with a special local_worker
# (such as SocketWorker)
# hook = TorchHook(verbose=False)
