21 U.S.C. § 332

1. (a)The district courts of the United States and the United States courts of the Territories shall have jurisdiction, for cause shown 1 So in original. Probably should be followed by a comma. to restrain violations of section 331 of this title, except paragraphs (h), (i), and (j).
2. (b)In case of violation of an injunction or restraining order issued under this section, which also constitutes a violation of this chapter, trial shall be by the court, or, upon demand of the accused, by a jury.