From b4b3d648868e4cec33c69e18fc3877c103a8d438 Mon Sep 17 00:00:00 2001
From: Hiroshiba Kazuyuki <kazuyuki_hiroshiba@dwango.co.jp>
Date: Mon, 1 Nov 2021 04:22:22 +0000
Subject: [PATCH] work on torch==1.10.0

---
 pytorch_trainer/dataset/convert.py | 2 +-
 setup.py                           | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pytorch_trainer/dataset/convert.py b/pytorch_trainer/dataset/convert.py
index 5103946a0..e30951728 100644
--- a/pytorch_trainer/dataset/convert.py
+++ b/pytorch_trainer/dataset/convert.py
@@ -3,7 +3,7 @@
 import numpy
 import six
 import torch
-from torch._six import container_abcs
+from collections import abc as container_abcs
 
 import pytorch_trainer
 
diff --git a/setup.py b/setup.py
index fcc49ddd2..5215d7752 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name='pytorch-trainer',
-    version='1.3.1',
+    version='1.3.3',
     packages=find_packages(),
     url='https://github.com/Hiroshiba/pytorch-trainer',
     author='Kazuyuki Hiroshiba',