QUAIL Syntax errors in 5-ml-synthetic-data

[joshua-oss]

https://github.com/opendp/smartnoise-samples/blob/master/whitepaper-demos/5-ml-synthetic-data.ipynb

In the cell with def QuailSynth(epsilon):

Complains because epsilon is now a required param. Adding epsilon to the PyTorchDPSynthesizer and the gan clears that error, but then the code dies with

~/opt/anaconda3/lib/python3.8/site-packages/torch/nn/modules/linear.py in init(self, in_features, out_features, bias)
74 self.in_features = in_features
75 self.out_features = out_features
---> 76 self.weight = Parameter(torch.Tensor(out_features, in_features))
77 if bias:
78 self.bias = Parameter(torch.Tensor(out_features))

TypeError: new(): argument 'size' must be tuple of ints, but found element of type float at pos