diff options
| author | peter <peterghost86@gmail.com> | 2019-03-13 09:07:57 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-03-13 09:22:46 -0700 |
| commit | 1c3494daf091022e0e5ff4c0bdea2bfef61a93c2 (patch) | |
| tree | 203dcf024cf058b515d3b7b7cc060c0651403252 /test/test_numba_integration.py | |
| parent | 9089182ce40d026efd8c4256fbf3e6db02865a85 (diff) | |
| download | pytorch-1c3494daf091022e0e5ff4c0bdea2bfef61a93c2.tar.gz pytorch-1c3494daf091022e0e5ff4c0bdea2bfef61a93c2.tar.bz2 pytorch-1c3494daf091022e0e5ff4c0bdea2bfef61a93c2.zip | |
Fix Windows test CI
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17954
Differential Revision: D14437473
Pulled By: soumith
fbshipit-source-id: f0d79ff0c5d735f822be3f42bbca91c1928dacaf
Diffstat (limited to 'test/test_numba_integration.py')
| -rw-r--r-- | test/test_numba_integration.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_numba_integration.py b/test/test_numba_integration.py index ba14916a8f..105e181a12 100644 --- a/test/test_numba_integration.py +++ b/test/test_numba_integration.py @@ -2,7 +2,7 @@ import unittest import sys import common_utils as common -from common_utils import TEST_NUMBA, TEST_NUMPY +from common_utils import TEST_NUMBA, TEST_NUMPY, IS_WINDOWS from common_cuda import TEST_NUMBA_CUDA, TEST_CUDA, TEST_MULTIGPU import torch @@ -117,7 +117,7 @@ class TestNumbaIntegration(common.TestCase): ] for dt in torch_dtypes: - if dt == torch.int8: + if dt == torch.int8 and not IS_WINDOWS: # "CharTensor" numpy conversion not supported with self.assertRaises(TypeError): torch.arange(10).to(dt).numpy() |