Commit babd9621 authored by peastman's avatar peastman
Browse files

More attempts at fixing tests

parent 85568f87
Loading
Loading
Loading
Loading
+5 −2
Original line number Diff line number Diff line
# -*- coding: UTF-8 -*-
from __future__ import division

import random
import string
from collections import Sequence
@@ -272,6 +270,11 @@ class Layer(object):
  def __neg__(self):
    return Multiply([self, Constant(-1.0)])

  def __div__(self, other):
    if not isinstance(other, Layer):
      other = Constant(other)
    return Divide([self, other])

  def __truediv__(self, other):
    if not isinstance(other, Layer):
      other = Constant(other)
+0 −1
Original line number Diff line number Diff line
@@ -149,7 +149,6 @@ class TestANIRegression(unittest.TestCase):
    assert self.model.layer_structures == restored_model.layer_structures
    assert self.model.atom_number_cases == restored_model.atom_number_cases
    assert self.model.batch_size == restored_model.batch_size
    assert self.model.learning_rate == restored_model.learning_rate
    assert self.model.use_queue == restored_model.use_queue

    assert expected == predicted