Some background
Let $G$ be a torsion-free nilpotent group of class $2$ and rank $2$ (i.e., generated by two elements). Then, $G$ has to be isomorphic to the Heisenberg group.
This is relatively easy to see: if $G$ has any relations besides those required for it to be nilpotent of class $2$, then such a relation can be expressed as $\lambda^k x^a y^b = 1$, where $G$ is generated by $x, y$ and $\lambda = [x, y]$. Thus, $y^{-b} = \lambda^k x^a \implies 1 = [\lambda^k x^a, y] = [x^a, y] = \lambda^a$. If $G$ is torsion-free, this forces $a = 0$, implying $y^{-b} = \lambda^k \implies 1 = [x, y^{-b}] = \lambda^{-b}$. Thus, $b$ is also equal to $0$ and $\lambda^k = 1 \implies k = 0$, meaning no such relation exists.
Question
Going one step further, the paper "A Note on Finitely Generated Torsion-Free Nilpotent Groups of Class $2$", by Fritz J. Grunewald and Rudolf Scharlau, the authors give multiple examples of torsion-free nilpotent groups of class $2$ and rank $4$ which are non-isomorphic, leading me to ask:
Does there exist any kind of classification of torsion-free nilpotent groups of class $2$? For any given rank, are there finitely many distinct isomorphism classes?
The paper I cited above constructs arbitrarily large collections of pairwise non-isomorphic torsion-free nilpotent groups of class $2$, but I'm not sure about their ranks...
I gratefully accept any references or databases, as I was unable to find much and, maybe I'm wrong, but I think GAP only has a finite group database...
Thanks in advance!