v2.0.1: Adjust some versions and docs.

kdexd · Mar 10, 2020 · 9c1b228 · 9c1b228
1 parent abf0efc
commit 9c1b228
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -23,9 +23,6 @@
 copyright = "2019, Karan Desai"
 author = "Karan Desai"
 
-# The full version, including alpha/beta/rc tags
-release = "1.0"
-
 
 # -- General configuration ---------------------------------------------------
 
@@ -61,9 +58,9 @@
 # built documents.
 #
 # This version is used underneath the title on the index page.
-version = "1.0.0"
+version = "2.0.1"
 # The following is used if you need to also include a more detailed version.
-release = "1.0.0"
+release = "2.0.1"
 
 # The language for content autogenerated by Sphinx. Refer to documentation
 # for a list of supported languages.

diff --git a/probnmn/trainers/_trainer.py b/probnmn/trainers/_trainer.py
@@ -3,7 +3,7 @@
 import torch
 from torch import nn, optim
 from torch.utils.data import DataLoader
-from torch.utils.tensorboard import SummaryWriter
+from tensorboardX import SummaryWriter
 
 from probnmn.config import Config
 from probnmn.utils.checkpointing import CheckpointManager

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-allennlp==0.8.0
+allennlp==0.9.0
 h5py==2.8.0
 loguru==0.4.1
 mypy_extensions==0.4.3